一个整型数组 nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n),空间复杂度是O(1)
首先我们考虑这个问题的一个简单版本:一个数组里除了一个数字之外,其他的数字都出现了两次。请写程序找出这个只出现一次的数字。
我们想到了异或运算的性质:任何一个数字异或它自己都等于0。也就是说,如果我们从头到尾依次异或数组中的每一个数字,那么最终的结果刚好是那个只出现一次的数字,因为那些出现两次的数字全部在异或中抵消掉了。
有了上面简单问题的解决方案之后,我们回到原始的问题。如果能够把原数组分为两个子数组。在每个子数组中,包含一个只出现一次的数字,而其他数字都出现两次。如果能够这样拆分原数组,按照前面的办法就是分别求出这两个只出现一次的数字了。
我们还是从头到尾依次异或数组中的每一个数字,那么最终得到的结果就是两个只出现一次的数字的异或结果。因为其他数字都出现了两次,在异或中全部抵消掉了。由于这两个数字肯定不一样,那么这个异或结果肯定不为0,也就是说在这个结果数字的二进制表示中至少就有一位为1。我们在结果数字中找到第一个为1的位的位置,记为第N位。现在我们以第N位是不是1为标准把原数组中的数字分成两个子数组,第一个子数组中每个数字的第N位都为1,而第二个子数组的每个数字的第N位都为0。
现在我们已经把原数组分成了两个子数组,每个子数组都包含一个只出现一次的数字,而其他数字都出现了两次。因此到此为止,所有的问题我们都已经解决。
简明扼要,做这个题三步走:
1.对所有数字异或,一样的数字抵消,出现一次的两个数字异或运算后必定不为 0;
2.用 lowbitlowbit 得到一个二进制位最右边一位为1的数字 maskmask,也就是两者出现不等的地方
3.maskmask 和数组的每个数字做与运算,等于 0 的分为一组,等于 maskmask 的分为一组,这同时也将两个不一样的数字分开了
class Solution {
public:
vector<int> singleNumbers(vector<int>& nums) {
int a = 0;
for(int i=0;i<nums.size();i++) a^=nums[i];
a = a & -a;
int b = 0, c = 0;
for(int i=0;i<nums.size();i++){
if(a&nums[i]) b^=nums[i];
else c^=nums[i];
}
vector<int> res({b,c});
return res;
}
};