LeeetCode442. 数组中重复的数据(巧用负号标记数组)

问题描述

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

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

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

示例：

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

输出:
[2,3]

思路

• 本题可以用哈希表存储每个元素的出现次数，但题目要求不使用额外的存储空间
• 由题可知，数组中的元素范围为1到n(n为数组长度)，而且每个数字最多出现两次
• 所以我们可以正负号表示是否出现过下标加1的数字，例如a[3]为负数则表示4出现过，a[0] > 0表示1没出现过
• 因此我们遍历一遍数组，将每一位上的值减1转换成下标，找此下标对应的值，如果为负数表示出现过，则将其下标加1作为结果添加到结果集。如果为正数就将其变为负数

代码

class Solution {
    public List<Integer> findDuplicates(int[] nums) {
        List<Integer> res = new ArrayList<>();
        for(int i = 0; i < nums.length; i++){
            int index = Math.abs(nums[i]) - 1; //数值转数组下标
            if(nums[index] < 0)  //对应的值是负数表示数字index + 1曾经出现过
                res.add(index + 1);
            else{
                
                 nums[index] *= -1;
            } 
            
        }
        return res;
    }
}