LeetCode刷题—使用计数数组

此类题一般有一个元素值为1~n 的数组，要求找到重复或缺失的数字。

448，找到所有数组中消失的数字，easy

给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。

找到所有在 [1, n] 范围之间没有出现在数组中的数字。

您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[5,6]

题解

题意：数组元素值为 1 ~ n，数组长度为n，有重复元素。

使用计数数组 count，从索引 1 存储 nums的元素值，count 元素值为0的下标即nums中缺失的元素。

代码

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
        int n = nums.length;
        List<Integer> res = new ArrayList<>();
        int[] count = new int[n + 1];
        //count数组的下标1~n用来存储nums数组元素1~n，count不含有的元素的下标即为没出现过的nums元素
        for(int i : nums){
            count[i]++;
        }
        for(int i = 1; i < n + 1; i++){
            if(count[i] == 0){
                res.add(i);
            }
        }
        return res;
    }
}

645，错误的集合，easy

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

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

示例 1:

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

注意:

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

题解

和上面题的思路一致，找到计数数组count中元素为 0 的即为缺失元素，为2的即为重复元素。

代码

class Solution {
    public int[] findErrorNums(int[] nums) {
        int n = nums.length;
        int[] count = new int[n + 1];
        int[] res = new int[2];
        for(int i : nums){
            count[i]++;
        }
        for(int i = 1; i < n + 1; i++){
            if(count[i] > 1){
                res[0] = i;
            }
            if(count[i] == 0){
                res[1] = i;
            }
        }
        return res;
    }
}

287，寻找重复数，medium

上面的双指针法已经是最优解法了，但计数数组的思想比较好想。

代码

class Solution {
    public int findDuplicate(int[] nums) {
    int n = nums.length;
        int[] count = new int[n + 1];
        for(int i : nums){
            count[i]++;
        }
        for(int i = 1; i < n + 1; i++){
            if(count[i] > 1){
                return i;
            }
        }
        return 0;
    }
}