环和杆【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; } }