从零学算法 (LCR 177. 撞色搭配)

整数数组 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 消除了,剩下为 xx0x 的 x 袜子
          // 把 xx1x 的同色袜子都异或给 y 消除了,剩下为 xx1x 的 y 袜子
          for(int n:nums){
              if((n&m)==0)x^=n;
              else y^=n;
          }
          return new int[]{x,y};
      }
    

你可能感兴趣的:(算法学习,#,位运算,算法,leetcode,数据结构)