1. 程式人生 > >Leetcode 442. 陣列中重複的資料

Leetcode 442. 陣列中重複的資料

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

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

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

示例:

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

輸出:
[2,3]

程式碼如下:

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