力扣-645. 错误的集合

问题描述

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

给定一个数组 nums 代表了集合 S 发生错误后的结果。

请你找出重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/set-mismatch

示例 1：

输入：nums = [1,2,2,4]
输出：[2,3]

示例 2：

输入：nums = [1,1]
输出：[1,2]

提示：

2 <= nums.length <= 104
1 <= nums[i] <= 104

思路:
在Python解法中，可利用set()的去重功能对重复数组进行去重。假如原列表为List_1 = [1,2,3,4,4]那么在进行set()去重操作之后就变为了set_1 = set(List_1) = {1,2,3,4}
此时可利用sum(List_1) - sum(set_1) = 4得到重复的元素。而可利用等差数列求和公式$$sum=\frac{n\ast (n+1)}{2}$$求得原始数列的和。sum - sum(set_1)即可得到缺失的那个元素。
代码实现

class Solution:
    def findErrorNums(self, nums: List[int]) -> List[int]:
        n = len(nums)
        total = n*(n+1)/2
        #求和
        num = int(total - sum(set(nums)))
        diff = int(total - sum(nums))
        return [num-diff,num]