【LeetCode】442. 数组中重复的数据

题目描述

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

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

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

示例：

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

输出:
[2,3]

 

我的思路（超时）

从第一个数开始查询，如果第一个数与第二个数相等，将nums[i]放入res，删除两个数；若不等，交换两个数位置，i++，查到最后，删除nums最后一个元素，从头开始，代码如下：

class Solution {
public:
    vector<int> findDuplicates(vector<int>& nums) {
        vector<int> res;
        int i = 0;
        while(i<nums.size()){
            if(nums.size() == 1){
                return res;
            }
            if(nums[i] == nums[i+1]){
                res.push_back(nums[i]);
                nums.erase(nums.begin()+i,nums.begin()+i+2);
                i = 0;
            }else{
                swap(nums[i],nums[i+1]);
                i++;
                if(i == nums.size()-1){
                    nums.pop_back();
                    i = 0;
                }
            }
        }
        return res;
    }
};

 

网上思路

给定一个整数数组 a，其中 1 ≤ a[i] ≤ n（n为数组长度）（这个条件非常关键！）

从第一个数开始遍历，对下标为 i 的数，将 nums[ i ] - 1 位置的元素置为负数，后续若出现相同数，那么其对应下标的数就为负数了，说明这个数出现了两次，将该数放入res数组。注意：每次取对应下标是应取绝对值（abs）。

class Solution {
public:
    vector<int> findDuplicates(vector<int>& nums) {
        vector<int> res;
        for(int i=0; i<nums.size(); i++){
            if(nums[abs(nums[i])-1] < 0)
                res.push_back(abs(nums[i]));
            else
                nums[abs(nums[i])-1] = - nums[abs(nums[i])-1] ;
        }
        return res;
    }
};