从零学算法 (LCR 177. 撞色搭配 / 题库 260)

与 260. 只出现一次的数字 III 相同
整数数组 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]

• 如果你对位运算有了一定的了解，你肯定会第一时间想到异或，因为异或运算有个特性就是 a^a = 0，所以就像连连看一样，遍历异或 [1,1,2,2,3] 以后会剩下 3。但是这一题就难在他并非只有一个单色袜，而是有两个，设他们为 x，y ，遍历异或完数组以后会剩下 x^y，记作 z，根据异或的特点可知，如果 z 的某一位为 1，那就表示 x 和 y 在那一位一个为 0 一个为 1。这也就是我们分出 x 和 y 的关键。我们就从右往左不断取 z 的某一位，看哪个是 1。用一个 m 记录，初始化为 0001，然后为 0010, 0100…。那么 x 和 y 与 m 进行 & 运算会得到不同结果，我们再遍历一遍数组就能以此得到 x 和 y 了。

•   public int[] sockCollocation(int[] nums) {
        int x=0,y=0,z=0,m=1;
        // 遍历异或得到 z
        for(int n:nums){
            z ^=n;
        }
        // 得到用来区分 x 和 y 的 m
        while((z&m)==0)m<<=1;
        // 反正最后和 x 异或的同色袜子会被消除，所以一直异或就行了
        // 比如 m 为 0010，就把 xx0x 的同色袜子都交给 x 消除了（0010&xx0x==0），剩下的就为 x 袜子
        // 否则把 xx1x 的同色袜子都异或给 y 消除了，剩下的为 y 袜子
        for(int n:nums){
            if((n&m)==0)x^=n;
            else y^=n;
        }
        return new int[]{x,y};
    }