USACO断开的项链(新技巧枚举)

你有一条由 N 个珠子串成的项链,珠子的颜色有红、白、蓝三种,珠子在项链中呈随机分布。

例如N=29 时,两个项链的示例如下所示:

            1 2                               1 2
        r b b r                           b r r b
      r         b                       b         b
     r           r                     b           r
    r             r                   w             r
   b               r                 w               w
  b                 b               r                 r
  b                 b               b                 b
  b                 b               r                 b
   r               r                 b               r
    b             r                   r             r
     b           r                     r           r
       r       r                         r       b
         r b r                             r r w
        样例 A                            样例 B
                    r 红珠子
                    b 蓝珠子
                    w 白珠子

图片对项链的第一颗珠子和第二颗珠子进行了标记。

样例 A 中的项链只包含蓝红两种颜色的珠子,将所有珠子按顺序记录颜色为:

brbrrrbbbrrrrrbrrbbrbbbbrrrrb

假设你现在要将项链从某个点处断开,并将断开后的项链拉直摆放,然后从一端开始收集相同颜色的珠子,直到碰到另一种颜色的珠子为止,完成后在另一端进行相同的操作(这次收集的珠子的颜色可能与之前收集的颜色并不相同)。

现在,你需要判断在项链的哪一处将项链断开,可以使得我们收集珠子的数量达到最多。

例如,对于样例 A 提供的项链,我们在 9 号珠子和 10号珠子之间断开项链,或者在 24 号珠子和 25 号珠子之间断开项链,可以收集到最多 8 个珠子。

另外,某些项链除蓝红珠子外,还包含白珠子,如样例 B 所示。

收集珠子时,如果我们遇到了白色珠子,那么我们可以将它视为红色或蓝色,并将其涂上相应的颜色。

表示项链的字符串只包含 r,w,b三种字符。

请你编写一个程序,求出我们可以收集珠子的最大数目。

输入格式
第一行包含整数 N,表示珠子的数量。

第二行包含一个由 N 个字符构成的字符串,字符串中只包含r,w,b 三种字符。

输出格式
共一行,包含一个整数表示我们可以收集珠子的最大数目。

数据范围
3≤N≤350
输入样例:
29
wwwbbrwrbrbrrbrbrwrwwrbwrwrrb
输出样例:
11
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 710;

int n;
char s[N];

int get(char c)
{
    
    
    if (c == 'b') return 1;//蓝色
    return 2;//红色
}

int main()
{
    
    
    cin >> n >> s;
    for (int i = 0; i < n; i ++ ) s[i + n] = s[i];//这个操作神奇

    int res = 0;
    for (int i = 0; i < n; i ++ )
    {
    
    
        int l = i, r = i + n - 1;//最多n个
        int cnt = 0, left = 0, right = 0;
        while (l <= r && (s[l] == 'w' || (left | get(s[l])) != 3))
        {
    
    
            if (s[l] != 'w') left |= get(s[l]);
            cnt ++, l ++ ;
        }
        while (l <= r && (s[r] == 'w' || (right | get(s[r])) != 3))
        {
    
    
            if (s[r] != 'w') right |= get(s[r]);
            cnt ++, r -- ;
        }
        res = max(res, cnt);
    }

    cout << res << endl;

    return 0;
}


猜你喜欢

转载自blog.csdn.net/qq_43738331/article/details/112756153