题目描述
给你一个长度为 n
的整数数组 nums
,其中 nums
的所有整数都在范围 [1, n]
内,且每个整数出现 一次 或 两次。请你找出所有出现 两次 的整数,并以数组形式返回。
你必须设计并实现一个时间复杂度为 O(n)
且仅使用常量额外空间的算法解决此问题。
样例
输入:nums = [4,3,2,7,8,2,3,1]
输出:[2,3]
输入:nums = [1,1,2]
输出:[1]
输入:nums = [1]
输出:[]
限制
n == nums.length
1 <= n <= 10^5
1 <= nums[i] <= n
- `nums 中的每个元素出现 一次 或 两次。
算法
(思维题) $O(n)$
- 我们遍历一次数组,每次将
abs(nums[i]) - 1
位置的数字变为相反数。若发现那个位置上的数字又变为正的时候,那个位置就是答案。
时间复杂度
- 仅扫描一遍数组,故时间复杂度为 $O(n)$。
空间复杂度
- 需要 $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;
}
};
vector[HTML_REMOVED] ans; 这个不算开了额外数组吗
返回值可以不计入空间复杂度
这都行吗
如果是不允许破坏原数组,该怎么做?
再复原一下就行了