题目描述
给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
解答
这种做法来自评论区。时间复杂度
,空间复杂度
。
做法:
因为数组元素已经规定了是在[1,n]之间,所以问题可以变得简单。数组大小是n,可能的元素一共有n个,即一个数组下标对于一个数。
以数组中的元素值减去1作为下标。如果对应的元素大于零,那么说明是第一次出现,乘以负一,标记这个下标对应元素已经出现过一次。如果下次再碰到为负数,那么就将元素保存。
注意:
这个题,已经说明了重复最多两次。如果是出现多次,那么要注意去重。
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
for(int i=0;i<nums.size();++i)
{
int k=abs(nums[i]);
if(nums[k-1]>0) nums[k-1]*=-1;
else if(nums[k-1]<0) res.push_back(k);
}
return res;
}
};