LeetCode 练习——645. 错误的集合

1.题目描述

645. 错误的集合

集合 s 包含从 1 到 n 的整数。不幸的是，因为数据错误，导致集合里面某一个数字复制了成了集合里面的另外一个数字的值，导致集合 丢失了一个数字 并且 有一个数字重复 。

给定一个数组 nums 代表了集合 S 发生错误后的结果。

请你找出重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

示例 1：

输入：nums = [1,2,2,4]
输出：[2,3]

示例 2：

输入：nums = [1,1]
输出：[1,2]

2.解题思路与代码

2.1 解题思路

这道题目两个要求，找到重复的数字，以及找到缺失的数字。输入数组原本是从 1 到 n 的连续数字，因此可以使用一个缓存数组来寻找重复的数字。使用长度和输入数组 nums 相同的 tmp 数组存放 nums 中每个数组出现的次数，当遇到 nums[i] 在 tmp 中不为 0 时，nums[i] 便是重复的那个数字。
找到重复的数字之后，再寻找缺失的数字。我们用求和公式求出原本应该的总和大小，然后再求出数组 nums 的实际总和大小，由于缺少一个元素，并用重复元素占位，因此缺失数字就等于原本总和大小减去实际总和大小再加上重复数字便是缺失的数字。

2.2 代码

class Solution {
    public int[] findErrorNums(int[] nums) {
        int[] ans = new int[2];
        int[] tmp = new int[nums.length + 1];
        int sum = 0;
        int target = (1 + nums.length) * nums.length / 2;
        for (int i = 0; i < nums.length; i++) {
            if (tmp[nums[i]] != 0) {
                ans[0] = nums[i];
            } else {
                tmp[nums[i]] = 1;
            }
            sum += nums[i];
        }
        ans[1] = target - sum + ans[0];
        return ans;

    }
}

2.3 测试结果

通过测试

3.总结

• 使用 tmp 数组存放 nums 中每个元素出现的次数
• 利用原本不缺失的数组总和减去实际数组总和再加上重复的数字，可以计算出缺失的数字