题目描述:
给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?
以任意顺序返回这两个数字均可。
示例 1:
输入: [1]
输出: [2,3]
示例 2:
输入: [2,3]
输出: [1,4]
思路 :我们将从1-n出现的数和nums中出现的数合并到一起,可以发现除了缺失的两个数(我们记为a,b)只出现了一次外,其余每个数都出现了两次,根据异或运算的性质,相同的数异或等于0,那么我们把这些数异或后得到的结果即为a和b异或的结果。
我们通过循环找到这两个数异或的结果中为1的比特位,然后把所有异或的数分成两组,两组之内分别异或得到最终的两个结果即为消失的数字
代码实现:
class Solution {
public:
vector<int> missingTwo(vector<int>& nums) {
//1.将所有数异或在一起
int tmp=0;
for(auto x:nums)
tmp^=x;
for(int i=1;i<=nums.size()+2;i++)
tmp^=i;
//2.找出a,b中比特位不同的那一位
int dif=0;
while(1)
{
if(((tmp>>dif)&1)==1)
break;
else dif++;
}
//3.根据dif位不同,将所有数划分为两类异或
int a=0,b=0;
for(int x:nums)
if(((x>>dif)&1)==1)
b^=x;
else
a^=x;
for(int i=1;i<=nums.size()+2;i++)
{
if(((i>>dif))&1==1)
b^=i;
else
a^=i;
}
return {a,b};
}
};