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;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值