力扣 645. 错误的集合位运算

最新推荐文章于 2022-05-18 15:52:10 发布

csu_xiji

最新推荐文章于 2022-05-18 15:52:10 发布

阅读量297

点赞数

分类专栏：力扣思维位运算

本文链接：https://blog.csdn.net/xiji333/article/details/105735816

版权

力扣同时被 3 个专栏收录

282 篇文章 2 订阅

订阅专栏

思维

181 篇文章 1 订阅

订阅专栏

位运算

24 篇文章 0 订阅

订阅专栏

https://leetcode-cn.com/problems/set-mismatch/

在这里插入图片描述

思路：做法很多，只考虑空间 $O (1)$ 的方法。设答案为 $a 、 b$ ，数组的异或和为 $s u m$ ，再对所有的 $1 < = i < = n$ 做一遍 $sum\ xor\ i$ ，得到的结果就是 $a\ xor\ b$ ，现在考虑怎么把他们分离出来。我们找到 $a\ xor\ b$ 的为 $1$ 的最低位，然后依据此位把原数组和 $[1, n]$ 分成两个部分，每个部分单独做一次异或和就可以得到 $a 、 b$ 了，现在只需要考虑它们的顺序问题，显然遍历一遍数组即可。

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        int siz=nums.size(),sum=0;
        for(int i=0;i<siz;i++)
            sum^=(i+1)^nums[i];
        int pos=1;
        while(!(sum&pos))
            pos<<=1;
        int a=0,b=0;
        for(int i=0;i<siz;i++)
        {
            if(nums[i]&pos)
                a^=nums[i];
            else
                b^=nums[i];
            if((i+1)&pos)
                a^=(i+1);
            else
                b^=(i+1);
        }
        vector<int> ans{a,b};
        for(int i=0;i<siz;i++)
        {
            if(nums[i]==b)
            {
                swap(ans[0],ans[1]);
                break;
            }
        }
        return ans;
    }
};