LeetCode 2103. 环和杆

总计有 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 。

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) {
        unordered_map<char, int> colorToBit{{'R', 1}, {'G', 2}, {'B', 4}};

        vector<int> ringVec(10);
        for (int i = 0; i < rings.size(); i += 2) {
            ringVec[rings[i + 1] - '0'] |= colorToBit[rings[i]];
        }

        int ans = 0;
        for (int i : ringVec) {
            if (i == 7) {
                ++ans;
            }
        }

        return ans;
    }
};

如果有n个环,此算法时间复杂度为O(n),空间复杂度为O(1)。

posted @   epiphanyy  阅读(10)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
历史上的今天:
2021-01-31 MySQL技术内幕InnoDB存储引擎 学习笔记 第四章 表
2021-01-31 LeetCode 1732. 找到最高海拔
2021-01-31 MySQL必知必会 学习笔记 第二十二章 使用视图
点击右上角即可分享
微信分享提示