leetcode | 448.找到所有数组中消失的数字

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)