【每日一题Day366】LC2103环和杆 | 状态压缩

环和杆【LC2103】

总计有 n 个环，环的颜色可以是红、绿、蓝中的一种。这些环分别穿在 10 根编号为 0 到 9 的杆上。

给你一个长度为 2n 的字符串 rings ，表示这 n 个环在杆上的分布。rings 中每两个字符形成一个 颜色位置对 ，用于描述每个环：

• 第 i 对中的 第一个 字符表示第 i 个环的 颜色 （'R'、'G'、'B'）。
• 第 i 对中的 第二个 字符表示第 i 个环的 位置 ，也就是位于哪根杆上（'0' 到 '9'）。

例如，"R3G2B1" 表示：共有 n == 3 个环，红色的环在编号为 3 的杆上，绿色的环在编号为 2 的杆上，蓝色的环在编号为 1 的杆上。

找出所有集齐 全部三种颜色 环的杆，并返回这种杆的数量。

• 思路

  使用状态压缩记录每个杆子的环颜色信息，如果有用三种颜色那么状态值为7；最后计数返回结果

• 实现

  class Solution {
      public int countPoints(String rings) {
          int[] mask = new int[10];
          Map<Character, Integer> map = new HashMap<>();
          map.put('R', 1);
          map.put('G', 2);
          map.put('B', 4);
          for (int i = 0; i < rings.length(); i += 2){
              int val = map.get(rings.charAt(i));
              int index = rings.charAt(i + 1) - '0';
              mask[index] |= val;
          }
          int res = 0;
          for (int i = 0; i < 10; i++){
              if (mask[i] == 7){
                  res++;
              }
          }
          return res;
      }
  }

  • 复杂度
    • 时间复杂度： O ( n ) \mathcal{O}(n) O(n)
    • 空间复杂度： O ( C ) \mathcal{O}(C) O(C)