记负法
出现在给一个数组的情况,数组中的数只出现一次或者两次,求出现两次的数字,或者未出现的数。对于数组下标[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;
}
};