力扣刷题Python笔记：找到所有数组中消失的数字

题目

给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。

找到所有在 [1, n] 范围之间没有出现在数组中的数字。

您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。

Python解法

我们通过遍历两次数组来解决该问题：

• 第一次遍历，我们假设数组中的元素是一个位置索引，例如元素1对应位置索引0、元素6对应位置索引5，以此类推，然后我们将对应位置索引的数组元素变为负数，那么遍历完这一遍之后数组中只有缺失数对应的位置索引处的元素为正；
• 第二次遍历，找到数组中正元素位置处对应的索引，然后加1将其放入结果数组即为最终结果。

代码如下：

def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
    res = []
    for i in nums:
        index = abs(i)-1
        if nums[index]>0:
            nums[index] *= -1
    for index, element in enumerate(nums):
        if element>0:
            res.append(index+1)
    return res