题目表述:给定一个整数数组 nums,其中恰好有两个元素只出现一次,其余所有元素均出现两次。 找出只出现一次的那两个元素。你可以按任意顺序返回答案。
例如
输入:nums = [1,2,1,3,2,5]
输出:[3,5]
解释:[5, 3] 也是有效的答案。
如果用哈希表来解决问题,那么这题就变成一道水题了,所以题目又加了一个条件:要求O(n)时间复杂度,O(1)的空间复杂度。
这道题的解法很巧妙,需要用到异或运算。先假设现在只有一个元素出现一次,剩余元素均出现两次,那么很简单,剩余的元素两两异或为0,结果就是只出现一次的元素(任何数跟0异或都等于其本身)。现在是求两个元素,怎么办?
思路是,把元素a和元素b分别分到A组和B组,剩余的元素成对分到A组或B组。这样A组的异或结果就是a,B组的异或结果就是b
如何进行分组?
设所有数的异或结果是xor,xor即为a和b的异或和。根据异或原理,当a和b的第x位不同时,xor的第x位为1。所以,我们取xor中不为0的最低位,作为分组的标准,如果为0分到A组,不为0则分到B组。对于两两出现的元素,肯定也会分在同一组。
代码如下
class Solution {
public int[] singleNumber(int[] nums) {
int xor = 0; //a和b异或结果
for(int n : nums){
xor ^= n;
}
int mask = xor & (-xor); //xor不为0的最低位
int[] ans = new int[2];
for(int n : nums){
if((mask & n) == 0){
ans[0] ^= n;
}
if((mask & n) != 0){
ans[1] ^= n;
}
}
return ans;
}
}