这是第18篇算法,力扣链接
整数数组
sockets
记录了一个袜子礼盒的颜色分布情况,其中sockets[i]
表示该袜子的颜色编号。礼盒中除了一款撞色搭配的袜子,每种颜色的袜子均有两只。请设计一个程序,在时间复杂度 O(n),空间复杂度O(1) 内找到这双撞色搭配袜子的两个颜色编号。示例 1:
输入:sockets = [4, 5, 2, 4, 6, 6] 输出:[2,5] 或 [5,2]示例 2:
输入:sockets = [1, 2, 4, 1, 4, 3, 12, 3] 输出:[2,12] 或 [12,2]
这道题有点像普通的那个找“单身狗”的题,那道题的做法是将数组的里的数组全部做异或运算,相同的会兑掉,但是在这道题里,有两个单身狗,做异或运算之后得到的是num1 和 num2的异或结果。
注意这里的结果为 num1 和 num2异或的结果,如果二进制位为1,证明这一位上 num1 和 num2 不同,我们可以利用这个规则,找到差异位,然后根据差异位去异或出两个数字。
代码如下:
func sockCollocation(sockets []int) []int {
result1, result2, helper, temp := 0, 0, 1, 0
for _, num := range sockets {
temp ^= num
}
for (temp & helper) == 0 {
helper = helper << 1
}
for _, num := range sockets {
if num&helper != 0 {
result1 ^= num
} else {
result2 ^= num
}
}
return []int{result1, result2}
}