给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?
以任意顺序返回这两个数字均可。
示例 1:
输入: [1]
输出: [2,3]
示例 2:
输入: [2,3]
输出: [1,4]
提示:
nums.length <= 30000
分析:先不看题目对时空复杂度的要求,单纯看这个题目,我们来构思一下
1.直接双层循环,第一层循环用于枚举1-N,第二层循环用于遍历数组,来检查哪两个数没有出现在该数组里。
2.第一种方法的复杂度无疑是很高的,因为一次循环数组的过程其实可以给我们很多信息,比如我们可以直接创建一个标记数组,每次遍历一个数就将对应位置的标记打上true,再遍历一遍标记数组就可以知道未出现的数是哪两个,复杂度为O(n),但需要O(n)的空间。
3.由于数字是1-N各出现一次,因此在不缺数的情况下,应该满足有且仅有i个数小于nums[i](i为从0开始的下标),那么很容易知道,如果数据有序,我们可以直接二分,根据条件:
A.i<nums[i],可知[0,i]的区间是不缺数的,接下来在区间[i+1,n-1]中查询
B.i=nums[i],可知[0,i]和[i+1,n-1]各缺一个数
C.i>nums[i](i=nums[i]+1,因为本题只缺两个数),可知[0,i]和[i+1,n-1]各缺一个数
当二分到区间边界时,就可以知道缺少的是哪个数了。
4.根据题目所说,O(n)的时间复杂度,O(1)的空间复杂度,如果做过找缺少的一个数就可以知道,本题在暗示使用抑或。
当缺少一个数的时候,只要用1-n的抑或结果再与所有存在的数抑或,就可以知道那个缺乏的数了。
假设本题缺少的数是num_one与num_two,如果按照抑或的思路,最后得到的结果是num_one^num_two。
然而由于num_one!=num_two,因此必有num_one^num_two!=0,抑或的运算法则是0^0=0、1^1=0、1^0=1,即两个不相等的数抑或后至少存在一个二进制数位为1,我们不妨取抑或结果的最低位即lowbit,此时我们知道num_one与num_two一个含有lowbit,一个不含有lowbit,同时1-n这些数字以及现有数组也可以按照是否含有lowbit分为两个部分,我们假设num_one含有lowbit,num_two不含有,那么显然num_one可以将数组以及1-n中含有lowbit的数全部抑或起来得到,num_two同理。
class Solution {
public:
vector<int> missingTwo(vector<int>& nums) {
int xor_res = 0, i, N = nums.size() + 2, lowbit, num_one = 0, num_two = 0;
for(int num : nums){
xor_res ^= num;
}
for(i = 1; i <= N; ++ i){
xor_res ^= i;
}
lowbit = xor_res & (-xor_res);
for(int num : nums){
if(num & lowbit){
num_one ^= num;
}else{
num_two ^= num;
}
}
for(i = 1; i <= N; ++ i){
if(i & lowbit){
num_one ^= i;
}else{
num_two ^= i;
}
}
return {num_one, num_two};
}
};