442. 数组中重复的数据

199 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

442. 数组中重复的数据

给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。

找到所有出现两次的元素。

你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?

示例:

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

输出:
[2,3]

解题思路

我们可以用一个哈希表记录数组 nums 中的数字,因为数字范围均在[1,n] 中,记录数字后我们再利用哈希表检查 [1,n] 中的每一个数是否出现,从而找到缺失的数字。

由于数字范围均在 [1,n] 中,我们也可以用一个长度为 n 的数组来代替哈希表

因此我们可以通过使用给定的原数组作为哈希表

具体实现

将nums[i]中的元素放入其对应的下标中,因为1 ≤ a[i] ≤ n,在没有出现重复的元素的情况下,应该是每一个元素都可以放在其对应的下标中,一旦出现重复的元素,就说明了会出现两个元素放入一个下标的情况

代码

class Solution {
    void swap(int[]nums,int l,int r)
    {
        int t=nums[r];
        nums[r]=nums[l];
        nums[l]=t;
    }
        public List<Integer> findDuplicates(int[] nums) {

            List<Integer> res=new ArrayList<>();
            for(int i=0;i<nums.length;i++)
            {
                  while(nums[i]>=0&&nums[i]!=i+1)
                  {
                    if(nums[nums[i]-1]==nums[i])
                    {
                        res.add(nums[i]);
                        nums[i]=-1;
                        break;
                    } 
                      swap(nums,i,nums[i]-1);
                  }

     
            }
            return res;
    }
}
  • 时间复杂度:O(n)。其中 n 是数组 nums 的长度。
  • 空间复杂度:O(1)。返回值不计入空间复杂度。