题目
一个整型数组 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
解题
主要思路:相比于上一题,此题要复杂一些;如果只是简单的进行异或运算,那么得到的值就是出现次数为1的两个元素的异或值,那么我们无法知道两个元素分别为多少。
如果可以把数组拆分成两个数组,并且出现1次的两个元素位于不同的数组内,那么分别对两个数组异或操作即可。那么难点在于满足要求的情况下,如何对数组进行分组?
在对数组所有值进行异或之后的值也就是两个出现1次的值的异或值,那么我们可以依据该异或值二进制中位值为1进行分组。简单点说,数组内,重复的数值最终都为0,只有出现1次的两个数进行异或操作,那么必然存在二进制下相同位上的数值(0或者1)不相等;根据这个就能对其进行分组。
时间复杂度:O(N) n为数组的长度
空间复杂度:O(1) 常量内存
public int[] singleNumbers(int[] nums) {
int k = 0;
for (int i = 0; i < nums.length; i++) {
k ^= nums[i];
}
int mask = 1;
while ((k & mask) == 0) {
mask <<= 1;
}
int a = 0, b = 0;
for (int i = 0; i < nums.length; i++) {
if ((nums[i]&mask)==0) {
a^=nums[i];
} else {
b^=nums[i];
}
}
return new int[]{a,b};
}