为什么我不能在 Python 中使用 max 来查找数组中的最大一位整数？

Question

我最近有一次面试，被要求用 Python 解决以下问题：

写一个函数：

def solution(A):

这样，给定一个由 N 个整数组成的数组A，它返回所有一位整数中的最大值。

例如，给定数组A如下：

[-6, -91, 1011, -100, 84, -22, 0, 1, 473]

函数应该返回 1。

假设：

• N 是 [1...1,000] 范围内的整数
• 数组A 的每个元素都是[-10,000..10,000] 范围内的整数。
• 数组A中至少有一个元素满足任务语句中的条件。

我想出了以下我认为很优雅的解决方案，但我得到的分数只有 55%：

def solution(A):
    return max([i for i in A if len(str(i))==1])

为什么这个解决方案不正确？

Answer 1

不是最优雅的。这假设例如-3 是个位数。

该算法将按照规则集进行。考虑约束：

• 数字是-10,000到10,000
• 有至少一个元素满足条件

然后我们可以将所有超过 9 的数字保护为最低的 -10,001，这样它们就不会被认为是最大的。我们不需要关心低于 -9 的数字，因为我们确信至少有 1 个元素（即单个数字，无论是正数还是负数，都大于那些低于 -9 的数字）满足条件.

print(
    max(
        [-6, -91, 1011, -100, 84, -22, 0, 1, 473, 5, 4, 67, 7, 3, 56],
        key=lambda num: -10001 if num > 9 else num
    )
)
print(
    max(
        [-6, -91, 1011, -100, 84, -22, 10, -5, -8, 67],
        key=lambda num: -10001 if num > 9 else num
    )
)

输出：

7
-5

Answer 2

这是一个在我的书中打勾的解决方案。

def solution(integers):
    """Return max single digit integer in sequence `integers`"""
    try:
        return max(i for i in integers if -9 <= i <= 9)
    except ValueError:
        return ValueError("Could not find single digit integer in list")

print(solution([-6, -91, 1011, -100, 84, -22, 0, 1, 473]))

它使用描述性参数名称，有注释，对数字大小进行简单比较，并使用带有内置函数 max 的生成器而不是不必要的列表。它还通过更热门的错误消息重新引发最预期的错误。该功能更易于维护和使用。