448.找到所有数组中消失的数字
题目描述
给你一个含 n 个整数的数组 nums ，其中 nums[i] 在区间 [1, n] 内。请你找出所有在 [1, n] 范围内但没有出现在 nums 中的数字，并以数组的形式返回结果。
示例 1：
输入：nums = [4,3,2,7,8,2,3,1]
输出：[5,6]
思路
可以通过集合求差集的方式实现
set1 = {4,3,2,7,8,2,3,1}
set2 = {1,2,3,4,5,6}
set2 - set1即为题目所求
代码
class Solution(object):
def findDisappearedNumbers(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
# 超出时间限制
# result = []
# for i in range(1, len(nums)+1):
# if i not in nums:
# result.append(i)
# return result
# 超出时间限制
# return [i for i in range(1, len(nums)+1) if i not in nums]
# 集合求差集
set1 = set(nums)
set2 = set([i for i in range(1, len(nums) + 1)])
return list(set2.difference(set1))
if __name__ == "__main__":
slt = Solution()
nums = [4, 3, 2, 7, 8, 2, 3, 1]
result = slt.findDisappearedNumbers(nums)
print(result)