题目
给定一个整数数组 nums,其中恰好有两个元素只出现一次,其余所有元素均出现两次。 找出只出现一次的那两个元素。
示例 :
输入: [1,2,1,3,2,5]
输出: [3,5]
注意:
结果输出的顺序并不重要,对于上面的例子, [5, 3] 也是正确答案。
你的算法应该具有线性时间复杂度。你能否仅使用常数空间复杂度来实现?
来源:力扣(LeetCode)
思路
首先,将其全部位进行 ‘异或’ ,我们知道相同的两个数 ‘异或’ 的话为0,那么数组中的1和2,都抵消掉了,就剩3和5 ‘异或’ 起来,那么就是二进制的 11 和 101 ‘异或’ ,得到110。
然后,进行 a &= -a 操作。首先变负数吧,在二进制中负数采用补码的形式,而补码就是反码 +1,那么 110 的反码是 11…1001,那么加1后是 11…1010,然后和 110 相与,得到了 10,就是代码中的 diff 变量。得到了 diff,就可以将原数组分为两个数组;
如果两个相同的数字 ‘异或’ ,每位都会是0,而不同的数字 ‘异或’ ,一定会有对应位不同,一个0一个1,这样 ‘异或’ 是1。
比如3和5的二进制 11 和 101,如果从低往高看,最开始产生不同的就是第二位,用第二位来和数组中每个数字相与,根据结果的不同,一定可以把3和5区分开来;
而其他的数字由于是成对出现,所以区分开来也是成对的,最终都会 ‘异或’ 成0,不会3和5产生影响;
分别将两个小组中的数字都异或起来,就可以得到最终结果。
C++代码
class Solution {
public:
vector<int> singleNumber(vector<int>& nums)
{
int diff = accumulate(nums.begin(), nums.end(), 0, bit_xor<int>());
diff &= -diff;
vector<int> res(2, 0);
for (auto &a : nums)
{
if (a & diff)
res[0] ^= a;
else
res[1] ^= a;
}
return res;
}
};