AC日记——[USACO1.1]坏掉的项链Broken Necklace 洛谷 P1203

题目描述

你有一条由N个红色的，白色的，或蓝色的珠子组成的项链(3<=N<=350)，珠子是随意安排的。 这里是 n=29 的二个例子:

第一和第二个珠子在图片中已经被作记号。

图片 A 中的项链可以用下面的字符串表示：

brbrrrbbbrrrrrbrrbbrbbbbrrrrb

假如你要在一些点打破项链,展开成一条直线，然后从一端开始收集同颜色的珠子直到你遇到一个不同的颜色珠子，在另一端做同样的事(颜色可能与在这之前收集的不同)。 确定应该在哪里打破项链来收集到最大数目的珠子。

例如，在图片 A 中的项链中，在珠子 9 和珠子 10 或珠子 24 和珠子 25 之间打断项链可以收集到8个珠子。

白色珠子什么意思?

在一些项链中还包括白色的珠子(如图片B) 所示。

当收集珠子的时候，一个被遇到的白色珠子可以被当做红色也可以被当做蓝色。

表现含有白珠项链的字符串将会包括三个符号 r ， b 和 w 。

写一个程序来确定从一条被给出的项链可以收集到的珠子最大数目。

输入输出格式

输入格式：

 

第 1 行: N, 珠子的数目

第 2 行: 一串长度为N的字符串, 每个字符是 r ， b 或 w。

 

输出格式：

 

输入输出样例

输入样例#1：

29 
wwwbbrwrbrbrrbrbrwrwwrbwrwrrb

输出样例#1：

11

说明

题目翻译来自NOCOW。

USACO Training Section 1.1

 

思路：

　　把序列处理成链，复制后接后面；

　　然后正着处理b，r，反着处理b，r；

　　然后枚举断点取最优值；

 

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

int n,len,ans,stlb[1001],stlr[1001],strb[1001],strr[1001];

char ch[1001];

int main()
{
    cin>>n;
    cin>>ch+1;
    for(int i=1;i<=n;i++) ch[i+n]=ch[i];
    len=strlen(ch+1);
    for(int i=1;i<=len;i++)
    {
        if(ch[i]=='b')
        {
            if(ch[i]==ch[i-1]||ch[i-1]=='w') stlb[i]=stlb[i-1];
            stlb[i]++;
        }
        else if(ch[i]=='r')
        {
            if(ch[i]==ch[i-1]||ch[i-1]=='w') stlr[i]=stlr[i-1];
            stlr[i]++;
        }
        else
        {
            stlr[i]=stlr[i-1]+1;
            stlb[i]=stlb[i-1]+1;
        }
    }
    for(int i=len;i>=1;i--)
    {
        if(ch[i]=='b')
        {
            if(ch[i]==ch[i+1]||ch[i+1]=='w') strb[i]=strb[i+1];
            strb[i]++;
        }
        else if(ch[i]=='r')
        {
            if(ch[i]==ch[i+1]||ch[i+1]=='w') strr[i]=strr[i+1];
            strr[i]++;
        }
        else
        {
            strr[i]=strr[i+1]+1;
            strb[i]=strb[i+1]+1;
        }
    }
    for(int i=1;i<=len;i++) ans=max(ans,max(stlr[i],stlb[i])+max(strr[i+1],strb[i+1]));
    cout<<min(ans,n);
    return 0;
}