18.撞色搭配

这是第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}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值