USACO Section 1.2 Broken Necklace

题目

题目描述

输入描述

Line 1: N, the number of beads
Line 2: a string of N characters, each of which is r, b, or w

输出描述

A single line containing the maximum of number of beads that can be collected from the supplied necklace.

样例输入

29
wwwbbrwrbrbrrbrbrwrwwrbwrwrrb

样例输出

11

题目分析

考虑到循环的情况，简单起见，直接把这个输入的字符串变成三个串黏在一起。考虑的串的范围是s[n]~s[2*n-1]
实际上就是求出每个点如是切下去，往左和往右各能延伸多少，取一个最大值。本题数据不是特别大，可能暴力并非不可能，我没有尝试。我的想法是一个个推出来。以ans[0][i]表示在第i的节点左边切下去，那么左边切口最长能延展多少；以ans[1][i]表示在第i的节点右边切下去，那么右边切口最长能延展多少。

推的过程

此处以ans[0][i]为例讲解。
首先第一个点ans[0][n]的求解:
先把s[n]左边所有的通用颜色w读掉并且记录数字，接着记录第一个出现的颜色字符到dif里面，接着继续往左扫，遇到w时长度显然是变长的（变色），遇到和dif内颜色相同的也显然变长，当遇到不一样的颜色时就中止，此时的最长长度就是ans[0][n]
接着往右边推了。请看我的灵魂作图

从左往右推的过程中，需要记录出现的最后一个非w字母是哪个字母，记在dif里面，wn记录连续出现了多少个w
当推到第i个位置的时候
如果i是w的话，那么直接s[i]=s[i-1]+1（因为可以变色黏上），之后wn++
如果i不是w且和dif相同的话，那么显然仍然是s[i]=s[i-1]+1 （中间的w变色成和他们一样）
如果i不是w且和dif不同的话，那s[i]=wn+1 （一串w变色成和i一样）
注意，在第二和第三种情况下，需要将wn置零并且更新dif是哪个颜色

需要避免的坑

注意，若是只有通用颜色w和其中某一种颜色比如说r的话，就会变成这样
wwbbbb
那么在最左边切开的话，往左往右最长都能延展6个长度，那么加起来的和就是12了，都超过串的总长了
另一种情况例如
wwbbrr，那么从bb rr这里切开的话，bb往左结合两个w长度变为4，rr往右结合两个w长度变为4，因此长度是8，也超过总长了。
这两种情况的最优解显然都是所有珠子都可以取到，因此若出现最大可能超过数量n的情况，答案就是n没错了

整体代码

/*
ID: penguin14
PROG: beads
LANG: C++
*/
#include<iostream>
#include <fstream>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
string s;
int ans[2][355*3];
int main() {
    ofstream fout("beads.out");
    ifstream fin("beads.in");
    int n, r, b, i;
    int cnt;
    int wn,maxx;
    char dif;
    bool flag;
    while (fin >> n) {
        flag = true;
        fin >> s;
        s = s + s + s;
        ans[0][n] = 1;
        i = n - 1;
        while (s[i] == 'w') {
            ans[0][n]++;
            i--;
        }
        dif = s[i];
        if (i != n - 1)
            s[n-1] = dif;
        for (i--;i>=0; --i) {
            if (s[i] == dif || s[i] == 'w') {
                ans[0][n]++;
            }
            else {
                break;
            }
        }

        ans[1][2*n-1] = 1;
        i = 2*n-1;
        while (s[i] == 'w') {
            ans[1][2*n-1]++;
            i++;
        }
        dif = s[i];
        if(i!=2*n-1)
            s[2*n-1]=dif;
        for(i++;i<3*n;++i){
            if(s[i]==dif||s[i]=='w'){
                ans[1][2*n-1]++;
            }
            else{
                break;
            }
        }

        wn = 0;
        dif = s[n-1];
        for(int i=n+1;i<2*n;++i){
            if(s[i-1]=='w'){
                wn++;
                ans[0][i]=ans[0][i-1]+1;
            }else{
                if(s[i-1]==dif){
                    ans[0][i]=ans[0][i-1]+1;
                }
                else{
                    ans[0][i]=wn+1;
                }
                wn=0;
                dif=s[i-1];
            }
        }

        wn=0;
        dif = s[2*n-1];
        for(int i=2*n-2;i>=n;--i){
            if(s[i]=='w'){
                wn++;
                ans[1][i]=ans[1][i+1]+1;
            }else{
                if(s[i]==dif){
                    ans[1][i]=ans[1][i+1]+1;
                }
                else{
                    ans[1][i]=wn+1;
                }
                wn=0;
                dif=s[i];
            }
        }

        maxx=ans[0][n]+ans[1][n];
        for(int a=n+1;a<2*n;a++){
            maxx=max(maxx,ans[0][a]+ans[1][a]);
            if(maxx>=n){
                fout<<n<<endl;
                flag = false;
                break;
            }
        }
        if(flag)
            fout<<maxx<<endl;
    }
    return 0;
}

USACO的题解

USACO指出我这个实际上是dp的做法
另外，最简单的版本，暴力是可行的