【C++】【USACO1.1.4】项链问题

【问题描述】

　　你有一条由N个红色的、白色的、或蓝色的珠子组成的项链，珠子是随意安排的。下面是 N=29 的两个项链：其中’r’代表 红色的珠子，’b’代表蓝色的珠子，’w’代表白色的珠子，项链中的第一和第二个珠子已经被作记号。比如图片 A 中的项链可以表示为：brbrrrbbbrrrrrbrrbbrbbbbrrrrb。
　　　　　　　　　
　 　　假如你要在一些点打破项链,展开成一条直线，然后从一端开始收集同颜色的珠子直到你遇到一个不同的颜色珠子，在另一端做同样的事(颜色可能与在这之前收集的不同)。 确定应该在哪里打破项链来收集到最大数目的珠子。例如，在图片 A 中的项链中，在珠子 9 和珠子 10 或珠子 24 和珠子 25 之间打断项链可以收集到8个珠子。在一些项链中还包括白色的珠子(如图片B)所示。当收集珠子的时候，一个被遇到的白色珠子可以被当做红色也可以被当做蓝色。写一个程序来确定从一条被给出的项链可以收集到的珠子最大数目。

【输入格式】

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

【输出格式】

单独的一行包含从被供应的项。

【输入样例】

29
wwwbbrwrbrbrrbrbrwrwwrbwrwrrb

【输出样例】

11

【数据范围】

3 <= N <= 350
【分析】
用数组bl,br,rl,rr分别记录在项链i处向左向右收集的篮色红色珠子数。
　　项链是环形的，但我们只要把两个同样的项链放在一块，就把它转换成线性的了【破环】。
　　补充：对w我们只需将它看成斗地主中的癞子即可，随便用
我们只要求出bl,br,rl,rr，那么结果就为
max(max(bl[i],rl[i])+max(br[i+1],rr[i+1])) (0<=i<=2*n-1)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,ok=0;
char a[400],b[800];
int bl[800],br[800],rl[800],rr[800];
int max_;

void run()
{
    bl[0]=rl[0]=0;//从左到右 
    for(int i=1;i<=2*n-1;i++)
    {
        if(b[i]=='r')   rl[i]=rl[i-1]+1,bl[i]=0;
        if(b[i]=='b')   bl[i]=bl[i-1]+1,rl[i]=0;
        if(b[i]=='w')   bl[i]=bl[i-1]+1,rl[i]=rl[i-1]+1;
    }
    br[2*n-1]=rr[2*n-1]=0;//从右到左 
    for(int i=2*n-2;i>=0;i--)
    {
        if(b[i]=='r')   rr[i]=rr[i+1]+1,br[i]=0;
        if(b[i]=='b')   br[i]=br[i+1]+1,rr[i]=0;
        if(b[i]=='w')   br[i]=br[i+1]+1,rr[i]=rr[i+1]+1;
    }   
}


int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d",&n);
    scanf("%s",a);
    strcpy(b,a);
    strcat(b+1,a);
    run();
    for(int i=0;i<2*n;i++)
    {
        max_=max(max_,max(bl[i],rl[i])+max(br[i+1],rr[i+1]));
        if(max_>n)  max_=n;
    }
    for(int i=0;i<2*n;i++)
    {
        if(b[i]=='w')   continue;
        else {ok=1;break;}
    }
    if(ok==1)
    cout<<max_;
    else
    cout<<n;
    return 0;
}