天天看點

442 Find All Duplicates in an Array 數組中重複的資料

給定一個整數數組 a,其中1 ≤ a[i] ≤ n (n為數組長度), 其中有些元素出現兩次而其他元素出現一次。

找到所有出現兩次的元素。

你可以不用到任何額外空間并在O(n)時間複雜度内解決這個問題嗎?

示例:

輸入:

[4,3,2,7,8,2,3,1]

輸出:

[2,3]

詳見:https://leetcode.com/problems/find-all-duplicates-in-an-array/description/

C++:

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