集合 S 包含从1到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个元素复制了成了集合里面的另外一个元素的值,导致集合丢失了一个整数并且有一个元素重复。
给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
示例 1:
输入: nums = [1,2,2,4]
输出: [2,3]
注意:
- 给定数组的长度范围是 [2, 10000]。
- 给定的数组是无序的。
分析:
题解一:
和 268. 缺失数字 挺相似,所以刚开始就想到和 268 类似的解法
利用 Set 内元素不重复的特性,找出重复的元素和不重复元素的和
再根据 1+2+…+n 的公式:(1+n)*n/2 ,求出原来的总和,减去上面求的总和就是差的值
class Solution {
public int[] findErrorNums(int[] nums) {
Set<Integer> set = new HashSet<>();
int[] ans = new int[2];
int sum = 0;
for (int i = 0; i < nums.length; i++) {
if (set.contains(nums[i]))
ans[0] = nums[i];
else {
set.add(nums[i]);
sum += nums[i];
}
}
ans[1] = (1 + nums.length) * nums.length / 2 - sum;
return ans;
}
}
题解二:
既然是 1-n,开个 n 容量的数组,存数出现的个数,如果出现 0 次的,就是缺的数,出现 2 次的,就是重复的数
class Solution {
public int[] findErrorNums(int[] nums) {
int[] bucket = new int[nums.length];
int[] ans = new int[2];
for (int i = 0; i < nums.length; i++) {
bucket[nums[i] - 1]++;
}
for (int i = 0; i < bucket.length; i++) {
if (bucket[i] == 2)
ans[0] = i + 1;
else if (bucket[i] == 0)
ans[1] = i + 1;
}
return ans;
}
}