Day14找出陣列中重複的資料
阿新 • • 發佈:2018-12-10
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; } };