python - 从列表中查找缺失的元素

Question

我编写这段代码是为了从给定列表中查找缺失的正整数，但在某些情况下它仍然返回 None。

例如，给定数组 A = [2, 4, 1, 5]，函数应返回 3，因为它是连续序列 (1 2 3 4 5) 中缺失的元素).

我做错了什么？

def solution(A):
    i = 1
    while i<len(A):
        if i not in A:
            return i
        i += 1

Answer 1

您需要注意 i < len(A)导致了一些问题。

给出这个例子:

A = [1, 2, 3, 5]

很明显，所需的答案是 4 , 但你的函数给出了 None .这是因为len(A) == 4因此，您的循环条件是 i < 4 ，它有效地枚举了 i从 1 到 3。

既然要找出丢失的号码，不妨在i时停止循环。达到列表中的最大数，而不是列表的长度，所以:

while i < max(A):

会是对的。