题目链接:LCR 177. 撞色搭配
题目
一个整型数组 nums
里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n)
,空间复杂度是O(1)
。
示例:
输入:nums = [4,1,4,6]
输出:[1,6] 或 [6,1]
解法
严格来说没有用到什么算法,更应该算是一种奇技淫巧。首先一个数本身做异或结果为0,0与任何数做异或得到这个数本身。因为数组中只有两个数不同,所以整个数组的异或等于这两个数的异或。
且这两个数的异或必有某一位是1,找出这一位,将数组中所有数根据这一位是否为1分为两类,相同的数必分在同一类,将这两类分别做异或即可得到这两个数。
ps:真是妙蛙种子吃着妙脆角进了米奇妙妙屋——妙到家了
补充:异或还有一个性质,A XOR B = C => C XOR B = A
代码
Java:
class Solution {
public int[] singleNumbers(int[] nums) {
int flag = 0;
for (int num : nums) {
flag ^= num;
}
int i = 0;
while (flag >> 1 != 0) {
flag = flag >> 1;
i++;
}
flag = 1 << i;
int[] ret = {0, 0};
for (int num : nums) {
if ((num & flag) == 0) {
ret[0] ^= num;
} else {
ret[1] ^= num;
}
}
return ret;
}
}