数组中寻找重复数字或缺失数字的记负法思想

记负法

出现在给一个数组的情况，数组中的数只出现一次或者两次，求出现两次的数字，或者未出现的数。对于数组下标

[0,n-1]

，数值的范围是

[1,n]

。

我们可以扫描一遍，将数字

i

的对应位置·

i-1

的值

nums[i-1]

置为负，即令

nums[i-1] = -nums[i-1]

， 这样，就不用额外的空间来记录某个数字是否出现。

用这样的方法，可以时间复杂度做到

O(n)

， 空间复杂度做到

O(1)

，极其优秀。

例题

1.442. Find All Duplicates in an Array

https://leetcode.com/problems/find-all-duplicates-in-an-array/description/

参考代码

class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
int len = nums.size(), val;
vector<int> ans;
for(int i = 0; i < len; i++) {
val = abs(nums[i])-1;
nums[val] = - nums[val];
if(nums[val] > 0) ans.push_back(val+1);
}
return ans;
}
};

2.448. Find All Numbers Disappeared in an Array

https://leetcode.com/problems/find-all-numbers-disappeared-in-an-array/description/

参考代码

class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int len = nums.size(), val;
for(int i = 0; i < len; i++) {
val = abs(nums[i])-1;
if(nums[val] > 0) {
nums[val] = -nums[val];
}
}
vector<int> ans;
for(int i = 0; i < len; i++) {
if(nums[i] > 0) {
ans.push_back(i+1);
}
}
return ans;
}
};