剑指offer—数组中数字出现的次数
题目描述
一个整型数组 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]
思路
假如题目要求是找出一个只出现一次的数,而其他都出现两次,那么直接遍历数组进行
互
相
异
或
\color{red}{互相异或}
互相异或就可以了
现在是需要找出两个只出现一次的数,而其他都出现两次,到底还能不能用异或来做呢?
当然,我们也可以继续遍历互相异或,最后得到的数是两个只出现一次数的异或的结果,有什么办法可以将它们分开吗?
1:首先,可以肯定的是这个结果值不为0,因为这两个数不相等,所以异或不为0(即二进制表示中肯定有1)
2:那异或结果的二进制中有1,这个1是由两个只出现一次的数的对应位置的比特位异或而来的(一个为0,一个为1)
3:那我们用这个位置的比特位再与每个数异或,==0为一边,==1为另一边,而其他出现两次的数肯定会分到同一边并且互相异或后为0,这样就可以将两个数分开,得到最终结果
代码
vector<int> singleNumbers(vector<int>& nums) {
vector<int> res(2);
int n = nums.size();
int add = 0;
for(int i = 0; i < n; i++){
add ^= nums[i];
}
//这里的掩码表示的是最低有效位的1
//正数取反+1为负数,那么正数和它的负数相与(&),得到的就是最低有效位的1
int mask = add & (-add);
res[0] = 0;
res[1] = 0;
for(int i = 0; i < n; i++){
//进行判断,这两个数这个掩码位上肯定一个为0,一个为1
//进行与操作,就可以将这两个数区分出来
if((mask & nums[i]) == 0){
res[0] ^= nums[i];
}else{
res[1] ^= nums[i];
}
}
return res;
}