2103. 环和杆

难度: 简单

来源: 每日一题 2023.11.02

---

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

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

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

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

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

示例 1：

输入：rings = "B0B6G0R6R0R6G9"
输出：1
解释：
- 编号 0 的杆上有 3 个环，集齐全部颜色：红、绿、蓝。
- 编号 6 的杆上有 3 个环，但只有红、蓝两种颜色。
- 编号 9 的杆上只有 1 个绿色环。
因此，集齐全部三种颜色环的杆的数目为 1 。

示例 2：

输入：rings = "B0R0G0R9R0B0G0"
输出：1
解释：
- 编号 0 的杆上有 6 个环，集齐全部颜色：红、绿、蓝。
- 编号 9 的杆上只有 1 个红色环。
因此，集齐全部三种颜色环的杆的数目为 1 。

示例 3：

输入：rings = "G4"
输出：0
解释：
只给了一个环，因此，不存在集齐全部三种颜色环的杆。

提示：

• rings.length == 2 * n
• 1 <= n <= 100
• 如 i 是 偶数 ，则 rings[i] 的值可以取 'R'、'G' 或 'B'（下标从 0 开始计数）
• 如 i 是 奇数 ，则 rings[i] 的值可以取 '0' 到 '9' 中的一个数字（下标从 0 开始计数）

class Solution {
    public int countPoints(String rings) {

    }
}

分析与题解

---

• HashMap记录法

  这个题目非常简单, 我们只需要通过HashMap记录每一根柱子上的颜色即可. 由于颜色色值可能重复, 所以我们使用 HashSet 作为Value. 利用它进行去重.

  HashMap<Character, Set<Character>> cache = new HashMap<>();
  

  当某一个柱子的颜色添加完当前的色值之后, Set的元素个数变成3个之后, 我们对结果数值进行 result++ 操作.

  if (colors.size() < 3) {
      colors.add(color);
      if (colors.size() == 3) {
          result++;
      }
      cache.put(point, colors);
  }
  

  那么接下来, 我们就看一下整体的题解过程.

  class Solution {
      public int countPoints(String rings) {
          HashMap<Character, Set<Character>> cache = new HashMap<>();
          int result = 0;
          for(int i = 0; i < rings.length(); i = i+2) {
              if(result >= 10) {
                  break;
              }
              Character color = rings.charAt(i);
              Character point = rings.charAt(i+1);
              Set<Character> colors = cache.getOrDefault(point, new HashSet<Character>());
              if (colors.size() < 3) {
                  colors.add(color);
                  if (colors.size() == 3) {
                      result++;
                  }
                  cache.put(point, colors);
              }
          }
          return result;
      }
  }
  

  复杂度分析:

  • 时间复杂度: O(n), 与字符串长度相关的时间复杂度.
  • 空间复杂度: O(n), HashMap与字符串长度相关的时间复杂度.

  结果如下所示.