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};} };