第十七个算法-数组判断重复

Posted 2021-03-05 liutian1912

给定一个整数数组 a，其中1 ≤ a[i] ≤ n （n为数组长度）, 其中有些元素出现两次而其他元素出现一次。找到所有出现两次的元素。

你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗？示例：输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]

由于数组元素具有1 ≤ a[i] ≤ n,这样的范围，那其实我们当每次经过一个元素时，给他加上n，
那些数值超过2n的元素索引+1，对应的就是我们的出现了两次的元素。

public List<Integer> findDuplicates(int[] nums) {
        List<Integer> list = new ArrayList<>();
        int n = nums.length;
        for(int i = 0; i < n; i++){
            int k = (nums[i] - 1) % n;
            nums[k] += n;
            if(nums[k]>2*n){
                list.add(k+1);
            }
        }
        return list;
    }

 

下面这个方法执行更快：

public List<Integer> findDuplicates(int[] nums) {
        List<Integer> res = new ArrayList<>();
        int[] counts = new int[nums.length + 1];
        for (int number : nums) {
            counts[number]++;
        }
        for (int i = 1; i < counts.length; i++) {
            if (counts[i] == 2)
                res.add(i);
        }
        return res;
    }

 

想法真奇妙。