题目
一个整型数组 nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n),空间复杂度是O(1)。
示例 1:
输入:nums = [4,1,4,6]
输出:[1,6] 或 [6,1]
示例 2:
输入:nums = [1,2,10,4,1,4,3,3]
输出:[2,10] 或 [10,2]
限制:
2 <= nums.length <= 10000
题目
看到题目要求空间复杂度为O(1),那就不能使用HashMap的方式,只能采取遍历时进行数组的标记。很容易想到从头到尾数字异或,但是题目有两个相同的数,异或出来是两个数的异或,那怎么样才能分开这两个数字呢。这里我们采用分组异或。如果我们可以把所有数字分成两组,使得:两个只出现一次的数字在不同的组中;**相同的数字会被分到相同的组中。**就可以做到每次异或的得到不同的值。具体怎么做呢?
记这两个只出现了一次的数字为 a 和 b,那么所有数字异或的结果就等于 a 和 b 异或的结果,我们记为x。如果我们把 x 写成二进制的形式x kx k−1⋯x 2 x1x0 ,其中 xi ∈{0,1},我们考虑一下xi=0 和xi=1 的含义是什么?它意味着如果我们把 a 和 b 写成二进制的形式,ai和 bi的关系——xi=1 表示 ai 和 bi不等,xi=0 表示ai 和 bi相等。假如我们任选一个不为 0的 xi,按照第 i位给原来的序列分组,如果该位为 0 就分到第一组,否则就分到第二组,这样就能满足以上两个条件。
代码
public int[] singleNumbers(int[] nums) {
int ret = 0;
for (int n : nums) {
ret ^= n;
}
int div = 1;
while ((div & ret) == 0) {
div <<= 1;
}
int a = 0, b = 0;
for (int n : nums) {
if ((div & n) != 0) {
a ^= n;
} else {
b ^= n;
}
}
return new int[]{a, b};
}
总结
遇到重复元素时想到位运算的性质,采取位运算来简化空间复杂度,同时巧妙利用位运算的特征。