442. Find All Duplicates in an Array
题目描述:
Given an array of integers, 1 ≤ a[i] ≤ n (n = size of array), some elements appear twice and others appear once.
Find all the elements that appear twice in this array.
Could you do it without extra space and in O(n) runtime?
Example:
Input: [4,3,2,7,8,2,3,1] Output: [2,3]意思就是给一个长度为n的数组,数组中的元素大小在[1,n]之间, 数组中的元素可能会重复出现两次,或者出现一次,或者不出现,找出所有重复出现两次的元素。
这一题和LeetCode的448题Find All Numbers Disappeared in an Array的思想是类似的,我在做Find All Numbers Disappeared in an Array这题时,用的是很一般的做法,在关于448题的讨论中,我看到了另一种时间复杂度为O(n),空间复杂度为O(1)的做法,并把它记录在了448题的博客中,根据这种思想,这题的时间复杂度为O(n),空间复杂度为O(1)的代码如下:
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> ans; //存储结果
for(int i=0;i<nums.size();i++){
int m=abs(nums[i])-1; //减一是为了让下标从0开始
if(nums[m]>0){
nums[m]=-nums[m];
}
else if(nums[m]<0){
ans.push_back(m+1);
}
}
return ans;
}
};
用时142ms,击败了60.12%的c++程序。
思想就是遍历数组一遍,在遍历的过程中,令m=abs(nums[i])-1,减一是为了让下标从0开始,如果nums[m]为正,就将nums[m]=-nums[m],说明m+1出现了一次;如果nums[m]为负,这说明m+1已经出现了一次,这次是第二次出现,所以将m+1放入ans中。