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

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

思路🧐：

  运用位运算，我们先创建一个变量tmp，依次异或nums和1~N所有整数，此时就可以得到两个数的异或值。

  然后通过右移操作，逐位检查tmp中的每一位，找到第一个为1的位，用diff变量存储起来，表示两个数字的该位是不同的(一个为0，一个为1，异或以后必定为1)。

  由此我们可以分组，diff位为1的为一组，diff位为0的为一组，分别对这两组数字与nums进行异或操作，得到两个的数字。再将这两个数字与1~N异或起来，而nums中的数字一定会出现两次，则相同数字异或结果为0，最后就可以得到两个缺失的数字。

代码🔎：

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int tmp = 0;
        for(auto e : nums) tmp ^= e;
        for(int i = 1; i <= nums.size() + 2; i++)    tmp ^= i; //得到缺失数字的异或
        int diff = 0; //找到不同位
        while(1)
        {
            if(((tmp >> diff) & 1) == 1) //如果该位为1，表示一个为0，一个为1，由此分组
                break;
            diff++;
        }
        int a = 0, b = 0; //由diff位是0是1进行分组
        for(auto e : nums)
            if(((e >> diff) & 1) == 1) a ^= e; // 该位为1，分到a组
            else b ^= e; //否则b组
        for(int i = 1; i <= nums.size() + 2; i++) // 与1~N所有数异或，也需要分组，不过此时相同的数异或为0，按照分组，最后得到两个缺失的数
            if(((i >> diff) & 1) == 1) a ^= i; // 该位为1，分到a组
            else b ^= i; //否则b组
        return {a,b};
    }
};