总计有 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:
def countPoints(self, rings: str) -> int:
mp = defaultdict(set)
for i in range(1, len(rings), 2):
c = int(rings[i])
mp[c].add(rings[i - 1])
return sum(len(v) == 3 for v in mp.values())
class Solution {
public int countPoints(String rings) {
Map<Integer, Set<Character>> mp = new HashMap<>();
for (int i = 1; i < rings.length(); i += 2) {
int c = rings.charAt(i) - '0';
mp.computeIfAbsent(c, k -> new HashSet<>()).add(rings.charAt(i - 1));
}
int ans = 0;
for (Set<Character> e : mp.values()) {
if (e.size() == 3) {
++ans;
}
}
return ans;
}
}
class Solution {
public:
int countPoints(string rings) {
unordered_map<int, unordered_set<char>> mp;
for (int i = 1; i < rings.size(); i += 2) {
int c = rings[i] - '0';
mp[c].insert(rings[i - 1]);
}
int ans = 0;
for (int i = 0; i < 10; ++i)
if (mp[i].size() == 3)
++ans;
return ans;
}
};
func countPoints(rings string) int {
mp := make(map[byte]map[byte]bool)
for i := 1; i < len(rings); i += 2 {
c := rings[i]
if len(mp[c]) == 0 {
mp[c] = make(map[byte]bool)
}
mp[c][rings[i-1]] = true
}
ans := 0
for _, v := range mp {
if len(v) == 3 {
ans++
}
}
return ans
}
function countPoints(rings: string): number {
const helper = (c: string) => c.charCodeAt(0) - 'A'.charCodeAt(0);
const n = rings.length;
const target = (1 << helper('R')) + (1 << helper('G')) + (1 << helper('B'));
const count = new Array(10).fill(0);
for (let i = 0; i < n; i += 2) {
count[rings[i + 1]] |= 1 << helper(rings[i]);
}
return count.reduce((r, v) => (r += v === target ? 1 : 0), 0);
}
impl Solution {
pub fn count_points(rings: String) -> i32 {
let rings = rings.as_bytes();
let target = (1 << b'R' - b'A') + (1 << b'G' - b'A') + (1 << b'B' - b'A');
let n = rings.len();
let mut count = [0; 10];
let mut i = 0;
while i < n {
count[(rings[i + 1] - b'0') as usize] |= 1 << rings[i] - b'A';
i += 2;
}
count.iter().filter(|&v| *v == target).count() as i32
}
}
int countPoints(char* rings) {
int target = (1 << ('R' - 'A')) + (1 << ('G' - 'A')) + (1 << ('B' - 'A'));
int count[10] = {0};
for (int i = 0; rings[i]; i += 2) {
count[rings[i + 1] - '0'] |= 1 << (rings[i] - 'A');
}
int ans = 0;
for (int i = 0; i < 10; i++) {
if (count[i] == target) {
ans++;
}
}
return ans;
}