LeetCode 645. 错误的集合（python）

题目链接

题目描述：

集合 S 包含从1到 n 的整数。不幸的是，因为数据错误，导致集合里面某一个元素复制了成了集合里面的另外一个元素的值，导致集合丢失了一个整数并且有一个元素重复。

给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

示例 1:

输入: nums = [1,2,2,4]
输出: [2,3]
注意:

给定数组的长度范围是 [2, 10000]。
给定的数组是无序的。

解题思路：

把集合s映射到列表res中，在数值相应的位置上出现一次就加1，遍历res，res中数值为2的索引为重复的整数，数值为0处的索引为丢失的整数。

class Solution(object):
    def findErrorNums(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        res=[0]*(len(nums)+1)
        for i in nums:
            res[i]+=1
        for i in range(1,len(res)):
            if res[i]==0:a=i
            if res[i]==2:b=i
        return [b,a]

p.s.别人的简洁的写法

return [sum(nums)-sum(set(nums)), sum(range(1, len(nums)+1)) - sum(set(nums))]