数组中数字出现的次数20210202
程序员文章站
2024-02-19 08:31:46
...
题目
一个整型数组 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次的两个元素位于不同的数组内,那么分别对两个数组异或操作即可。那么难点在于满足要求的情况下,如何对数组进行分组?
???
时间复杂度: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};
}