题目描述
给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
我的思路(超时)
从第一个数开始查询,如果第一个数与第二个数相等,将nums[i]放入res,删除两个数;若不等,交换两个数位置,i++,查到最后,删除nums最后一个元素,从头开始,代码如下:
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
int i = 0;
while(i<nums.size()){
if(nums.size() == 1){
return res;
}
if(nums[i] == nums[i+1]){
res.push_back(nums[i]);
nums.erase(nums.begin()+i,nums.begin()+i+2);
i = 0;
}else{
swap(nums[i],nums[i+1]);
i++;
if(i == nums.size()-1){
nums.pop_back();
i = 0;
}
}
}
return res;
}
};
网上思路
给定一个整数数组 a,其中 1 ≤ a[i] ≤ n(n为数组长度)(这个条件非常关键!)
从第一个数开始遍历,对下标为 i 的数,将 nums[ i ] - 1 位置的元素置为负数,后续若出现相同数,那么其对应下标的数就为负数了,说明这个数出现了两次,将该数放入res数组。注意:每次取对应下标是应取绝对值(abs)。
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
for(int i=0; i<nums.size(); i++){
if(nums[abs(nums[i])-1] < 0)
res.push_back(abs(nums[i]));
else
nums[abs(nums[i])-1] = - nums[abs(nums[i])-1] ;
}
return res;
}
};