描述
给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
思路
如果用空间的话,我们可以用hash表解决。这里不可以用额外空间,那我们只能充分利用这个数组本身的空间咯。
数组元素全部是1到n之间的,找到出现两次的元素,我们可以将数组元素当做索引来看待,有的索引出现一次有的出现两次,如果我们将索引位置的元素取反,出现两次的就会被取反两次,索引位置的值不变,所以如果索引位置的元素还是正数的话就表示被取反两次。
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
for(int a:nums){
int idx=abs(a)-1;
nums[idx]*=(-1);
if(nums[idx]>0)res.push_back(idx+1);
}
return res;
}
};