集合 s 包含从 1 到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个数字复制了成了集合里面的另外一个数字的值,导致集合 丢失了一个数字 并且 有一个数字重复 。
给定一个数组 nums 代表了集合 S 发生错误后的结果。
请你找出重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
示例 1:
输入:nums = [1,2,2,4]
输出:[2,3]
示例 2:
输入:nums = [1,1]
输出:[1,2]
提示:
2 <= nums.length <= 104
1 <= nums[i] <= 104
class Solution:
def findErrorNums(self, nums: List[int]) -> List[int]:
n = len(nums)
S = sum(set(nums))
RepeatedNum = sum(nums)-S
ReplacedNum = n*(n+1)//2-S
return [RepeatedNum, ReplacedNum]
使用等差数列求和公式,求出n个数的和,减去去掉重复数字的列表的和就是缺失的数,巧妙的方法来源于评论区。等差数列也可替换为
sum(range(len(nums)+1))