目录

0645:错误的集合

力扣第 645 题

题目

集合 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

相似问题:

分析

去重后显然元素之和少的就是重复的数。而去重后的元素之和比 1 到 n 之和少的就是丢失的数字。

解答

1
2
3
def findErrorNums(self, nums: List[int]) -> List[int]:
	s, n = sum(set(nums)), len(nums)
	return [sum(nums) - s, n*(n+1)//2 - s]

48 ms