leetcode_2103 环和杆

1. 题意

10个环。给定一个字符串每两个字符一组，第一个字符表示颜色(R,G,B),第二个字符表示在第几根杆上。求多少根杆上每个颜色都有。

环和杆

2. 题解

直接模拟记录即可，用位运算压缩。

class Solution {
public:
    int countPoints(string rings) {
        function<int(char)> getIdx = [](char c)->int
        {
            if ( c == 'R')
                return 1;
            if ( c == 'G')
                return 2;
            if ( c == 'B' )
                return 4;
            return 0;
        };

        int sz = rings.size();

        int mem[10] = {0};
        for ( int i = 0;i < sz; i += 2) {
            mem[rings[i + 1] - '0'] |= getIdx(rings[i]);
        }


        return count(mem, mem + 10, 7);
    }

};