给你一个含 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.length1 <= n <= 1051 <= nums[i] <= n
进阶:你能在不使用额外空间且时间复杂度为 O(n) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
实现:
可以使用原地修改数组的方法,在遍历 nums 数组时,将出现的数字作为下标,将对应位置的数字取相反数,即 nums[abs(nums[i])-1] = -abs(nums[abs(nums[i])-1])。
遍历完成后,再次遍历数组,将为正数的下标加 1 加入到结果数组中即可。
时间复杂度为 O(n),空间复杂度为 O(1)。
以下是 Python 代码实现:
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
res = []
for i in range(len(nums)):
index = abs(nums[i]) - 1
if nums[index] > 0:
nums[index] = -nums[index]
for i in range(len(nums)):
if nums[i] > 0:
res.append(i+1)
return res
浙公网安备 33010602011771号