位运算:消失的两个数字

题目描述:

给定一个数组,包含从 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 missingTwo(vector& 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};
    }
};

你可能感兴趣的:(leetcode,算法,职场和发展)