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]想到另外一题:如果是只有一个出现1次的数,其他都出现2次,可以直接异或所有数,结果就是那个数
常规做法用hash table 如果当前数字对应的值已经是1,则pushback,代码如下:
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
unordered_map<int, int> hash;
vector<int> res;
for (int i = 0; i < nums.size(); i++) {
if (hash[nums[i]] == 1) {
res.push_back(nums[i]);
}
hash[nums[i]]++;
}
return res;
}
};
但是要求空间复杂度O(1),想到用本来的数组做hash table,用index表示数字的值,因为数字是从1-n,而index从0开始,所以用nums[i]-1表示index,把当前index对应的数字取负,如果遍历到同一个值,会map到同一个index,如果此时对应值已经为负数,则说明遍历到相同值了。
注意index = abs(nums[i])-1因为nums[i]可能已经被取反了。
代码如下:
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int> res;
for (int i = 0; i < nums.size(); i++) {
int index = abs(nums[i]) - 1;//记得加abs,因为数值可能已经被取反了
if (nums[index] < 0) {
res.push_back(index + 1);
}
else {
nums[index] = -nums[index] ;
}
}
return res;
}
};