题目
645. 错误的集合
https://leetcode-cn.com/problems/set-mismatch/
集合 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减去总和就是差值,再加上重复数字就是丢失的数字。
代码
/**
* @param {number[]} nums
* @return {number[]}
*/
var findErrorNums = function (nums) {
const hash = new Set();
const n = nums.length;
const ans = [];
let sum = 0;
// 判断是否找到重复数字
let flag = false;
for (let i = 0; i < n; i++) {
if (hash.has(nums[i])) {
// 找到重复数组
ans.push(nums[i]);
flag = true;
} else if(!flag){
hash.add(nums[i]);
}
sum += nums[i];
}
// 1到n相加后减去重复数字就是丢失的数字
ans.push(n * (n + 1) / 2 - sum + ans[0]);
return ans;
};