题目:力扣
给你一个含 n 个整数的数组 nums ,其中 nums[i] 在区间 [1, n] 内。请你找出所有在 [1, n] 范围内但没有出现在 nums 中的数字,并以数组的形式返回结果。
示例 1:
输入:nums = [4,3,2,7,8,2,3,1]
输出:[5,6]
示例 2:
输入:nums = [1,1]
输出:[2]
提示:
n == nums.length
1 <= n <= 105
1 <= nums[i] <= n
进阶:你能在不使用额外空间且时间复杂度为 O(n) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
解题思路:
用当前数组下标+1表示元素,如果数组出现该元素,则数组下标所对应的数+n,最后遍历数组找出小于n+1的数即为没出现过的元素。
注意不能直接置0,否则会影响后续元素的判断(+n后%n还能还原值)
代码:
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
n = len(nums)
for i in nums:
x = i % n - 1
nums[x] = nums[x] + n
res = []
for i, x in enumerate(nums):
if x < n + 1:
res.append(i+1)
return res