给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
解题思路:由于所有元素限定在1到n之间,所以可以先交换第i个元素与第nums[i] - 1个元素的位置,然后遍历nums数组判断是否与[1....n]一致,不一致表明元素重复;
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
for (int i = 0; i < nums.size(); ) {
if (nums[i] != nums[nums[i] - 1]) {
swap(nums[i], nums[nums[i] - 1]);
} else {
i++;
}
}
for (int j = 0; j < nums.size(); j++) {
if ((j + 1) != nums[j]) {
res.push_back(nums[j]);
}
}
return res;
}
};