LeetCode.645. 错误的集合

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

给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。

示例 1:

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

注意:

  1. 给定数组的长度范围是 [2, 10000]。
  2. 给定的数组是无序的。

分析:
题解一:
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;
    }
}

猜你喜欢

转载自blog.csdn.net/liyuanyue2017/article/details/81270182