第十八周作业:
442. Find All Duplicates in an Array
解题思路:
Given an array of integers, 1 ≤ a[i] ≤ n (n = size of array), some elements appear twice and others appear once.
Find all the elements that appear twice in this array.
Could you do it without extra space and in O(n) runtime?
Example:
Input: [4,3,2,7,8,2,3,1] Output: [2,3]
思路:由于元素只能出现一次或者两次,所以我们不用担心过多元素占用位置,可以放心使用当前的值作为索引。采用交换位置法,将元素的值放到对应的位置。如果有某个值出现了两次,那么这个值一定会被先被换到比较靠前的位置,这时我们会发现它应该所在的位置已经有元素了,因此我们就将它扔在前边。再计算结果集的时候,如果一个元素出现在它的索引更靠前的位置,我们就可以认为它是一个冗余元素,将它加入索引集即可。
代码如下:
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> result;
int i = 0;
while (i < nums.size()) {
if (nums[i] != nums[nums[i] - 1]) {
swap(nums[i], nums[nums[i] - 1]);
} else {
++i;
}
}
for (i = 0; i < nums.size(); ++i) {
if (i != nums[i] - 1) {
result.emplace_back(nums[i]);
}
}
return result;
}
};