面试题 17.19. 消失的两个数字

思路

这段代码是解决一个数组中缺失两个数的问题。主要思路是利用异或运算的性质来找到缺失的两个数。先将数组中的数和 [1, n + 2] 区间内的所有数「异或」在⼀起，问题就变成了：有两个数出现了「⼀次」，其余所有的数出现了「两次」。

解题方法

第一个 for 循环中，通过对数组中的所有元素进行异或操作，找出数组中所有元素的异或结果，因为相同的数异或结果为0，而缺失的两个数则会留下。
第二个 for 循环中，从 1 到数组长度加 2（因为缺失了两个数），对这些数进行异或操作，得到的结果包含了缺失的两个数。
然后通过异或结果中不同比特位的位置，将数组中的元素分成两类。对于每一类，分别异或操作，即可得到缺失的两个数。

复杂度

时间复杂度: O(n)

空间复杂度: O(1)

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int tmp = 0;
        for(auto x : nums) tmp ^= x;
        for(int i = 1; i <= nums.size() + 2; i++) tmp ^= i;
        // 2. 找出 a，b 中⽐特位不同的那⼀位
        int diff = 0;
        while(1)
        {
            if(((tmp >> diff) & 1) == 1) break;
            else diff++;
        }
        // 3. 根据 diff 位的不同，将所有的数划分为两类来异或
        int a = 0, b = 0;
        for(int x : nums)
        if(((x >> diff) & 1) == 1) b ^= x;
        else a ^= x;
        for(int i = 1; i <= nums.size() + 2; i++)
        if(((i >> diff) & 1) == 1) b ^= i;
        else a ^= i;
        return {a,b};
    }
};

//