
题目
找到所有数组中消失的数字
给你一个含 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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array/
解
看题目,第一反应就是直接遍历1-n,发现这个数不在nums里面就追加到结果数组。
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
res = []
for i in range(1, len(nums) + 1):
if i not in set(nums):
res.append(i)
return res
结果超时。

仔细检查代码,看下为啥会超时。 第5行, 这个set(nums)放在了for循环里面,就会导致每次循环都会去set一下,这样就会导致时间超长。
修改一下。
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
res = []
n_nums = set(nums)
for i in range(1, len(nums) + 1):
if i not in n_nums:
res.append(i)
return res

它山之石
还有个解法--原地修改。
主要是这题的特殊性, 数组的长度确认为n。这样就可以将数组的值-1作
为索引,然后对该索引指向的值进行*-1处理。需要注意若当前值已经为负时,
需要跳过。这样处理一轮之后,数组中大于0的数对应的索引值+1就是我们需要
的结果。
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
res = []
for num in nums:
if nums[abs(num) - 1] > 0:
nums[abs(num) - 1] *= -1
for idx, num in enumerate(nums):
if num > 0:
res.append(idx + 1)
return res

文章转载自测试开发吃货,如果涉嫌侵权,请发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。




