题目描述
给定一个整数数组,其中1 ≤ a[i] ≤ n(n 为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以用 $O(1)$ 的额外空间并在 $O(n)$ 时间复杂度内解决这个问题吗?
样例
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
算法
(技巧) $O(n)$
- 我们遍历一次数组,每次将
abs(nums[i]) - 1
位置的数字变为相反数。若发现那个位置上的数字又变为正的时候,那个位置就是答案。
时间复杂度
- 仅扫描一遍数组,故时间复杂度为 $O(n)$。
空间复杂度
- 除了记录答案外,只需常数空间。
C++ 代码
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
int n = nums.size();
vector<int> ans;
for (int i = 0; i < n; i++) {
int pos = abs(nums[i]) - 1;
nums[pos] = -nums[pos];
if (nums[pos] > 0)
ans.push_back(pos + 1);
}
return ans;
}
};
如果是不允许破坏原数组,该怎么做?
再复原一下就行了