Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
一、题目描述:
645. 错误的集合 - 力扣(LeetCode) (leetcode-cn.com)
集合 s 包含从 1 到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个数字复制了成了集合里面的另外一个数字的值,导致集合 丢失了一个数字 并且 有一个数字重复 。
给定一个数组 nums 代表了集合 S 发生错误后的结果。
请你找出重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
示例 1:
输入:nums = [1,2,2,4]
输出:[2,3]
示例 2:
输入:nums = [1,1]
输出:[1,2]
提示:
- 2 <= nums.length <= 10^4
- 1 <= nums[i] <= 10^4
二、思路分析:
-
找出重复的元素。由题意可知,只有重复元素出现了两次,其余元素都只出现一次。因此我们可以使用根据sum(nums)-sum(set(nums))求出重复元素。
-
找出缺失元素。因为集合里面某一个数字复制了成了集合里面的另外一个数字的值,因此这个元素与set(nums)中元素正好构成前n个自然数。所以使用len(nums)*(len(nums)+1)//2求前n项数字和,再减去sum(set(nums))即可得到缺失的那一项。
三、AC 代码:
class Solution:
def findErrorNums(self, nums: List[int]) -> List[int]:
return [sum(nums)-sum(set(nums)), (1+len(nums))*len(nums)//2-sum(set(nums))]