题目
难度:★☆☆☆☆
类型:数组
集合 S 包含从1到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个元素复制了成了集合里面的另外一个元素的值,导致集合丢失了一个整数并且有一个元素重复。
给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
注意
给定数组的长度范围是 [2, 10000]。
给定的数组是无序的。
示例
输入: nums = [1,2,2,4]
输出: [2,3]
解答
我们可以分别求取重复的元素和缺失的元素,这里由于只有一个重复元素,我们可以将输入数组之和与去重后数组之和做差求取重复元素,对于缺失元素,我们可以将补全后的集合与去重后的集合做差求取。
class Solution:
def findErrorNums(self, nums: list) -> list:
nums_set = set(nums) # 获得输入数组的集合
repeat = sum(nums) - sum(nums_set) # 求和相减即为重复
correct_set = set(range(1, len(nums)+1)) # 理想情况下的数组
defect = correct_set - nums_set # 集合相减即为缺失
return [repeat, defect.pop()] # 返回结果
如有疑问或建议,欢迎评论区留言~