1. 程式人生 > >Day14找出陣列中重複的資料

Day14找出陣列中重複的資料

problem describe: 給定一個整數陣列 a,其中1 ≤ a[i] ≤ n (n為陣列長度), 其中有些元素出現兩次而其他元素出現一次。 找到所有出現兩次的元素。 你可以不用到任何額外空間並在O(n)時間複雜度內解決這個問題嗎? solution: 因為題目限定了nums[i]的大小在1-n之間,所以我們只需改動給定的陣列nums,將元素x存入nums[x-1]之中,最後檢查一遍陣列,不滿足條件的就是多出來的重複數

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