14231: 【原4231】Broken Necklace

题目

题目描述

author: Ge Yan 原OJ链接：https://acm.sjtu.edu.cn/OnlineJudge-old/problem/4231

Description

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

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

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

brbrrrbbbrrrrrbrrbbrbbbbrrrrb

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

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

白色珠子什么意思?

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

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

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

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

Input Format

第 1 行: N, 珠子的数目

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

Output Format

输出一行一个整数，表示从给出的项链中可以收集到的珠子的最大数量。

Sample Input

29 
wwwbbrwrbrbrrbrbrwrwwrbwrwrrb

Sample Output

11

ligongzzz's solution

#include "iostream"
#include "cstdio"
#include "cstring"
using namespace std;

char color_data[400] = { 0 };
int red_maxl[400] = { 0 }, red_maxr[400] = { 0 },
blue_maxl[400] = { 0 }, blue_maxr[400] = { 0 };

int mmax(int a, int b) {
    return a > b ? a : b;
}

int main() {
    int n;
    scanf("%d", &n);
    scanf("%s", color_data);

    //统计初始位置
    if (color_data[n - 1] == 'w') {
        for (int i = 0; i < n && (color_data[i] == 'w' || color_data[i] == 'r'); ++i)
            ++red_maxr[n - 1];
        for (int i = 0; i < n && (color_data[i] == 'w' || color_data[i] == 'b'); ++i)
            ++blue_maxr[n - 1];
    }
    else if (color_data[n - 1] == 'r') {
        for (int i = 0; i < n && (color_data[i] == 'w' || color_data[i] == 'r'); ++i)
            ++red_maxr[n - 1];
        blue_maxr[n - 1] = -1;
    }
    else {
        for (int i = 0; i < n && (color_data[i] == 'w' || color_data[i] == 'b'); ++i)
            ++blue_maxr[n - 1];
        red_maxr[n - 1] = -1;
    }

    if (color_data[0] == 'w') {
        for (int i = n - 1; i >= 0 && (color_data[i] == 'w' || color_data[i] == 'r'); --i)
            ++red_maxl[0];
        for (int i = n - 1; i >= 0 && (color_data[i] == 'w' || color_data[i] == 'b'); --i)
            ++blue_maxl[0];
    }
    else if (color_data[0] == 'r') {
        for (int i = n - 1; i >= 0 && (color_data[i] == 'w' || color_data[i] == 'r'); --i)
            ++red_maxl[0];
        blue_maxl[0] = -1;
    }
    else {
        for (int i = n - 1; i >= 0 && (color_data[i] == 'w' || color_data[i] == 'b'); --i)
            ++blue_maxl[0];
        red_maxl[0] = -1;
    }

    //递归统计
    for (int i = 1; i < n; ++i) {
        if (color_data[i] == 'w') {
            red_maxl[i] = red_maxl[i - 1] + 1;
            blue_maxl[i] = blue_maxl[i - 1] + 1;

        }
        else if (color_data[i] == 'r') {
            red_maxl[i] = red_maxl[i - 1] + 1;
            blue_maxl[i] = -1;
        }
        else {
            blue_maxl[i] = blue_maxl[i - 1] + 1;
            red_maxl[i] = -1;
        }
    }
    for (int i = n - 2; i >= 0; --i) {
        if (color_data[i] == 'w') {
            red_maxr[i] = red_maxr[i + 1] + 1;
            blue_maxr[i] = blue_maxr[i + 1] + 1;
        }
        else if (color_data[i] == 'r') {
            red_maxr[i] = red_maxr[i + 1] + 1;
            blue_maxr[i] = -1;
        }
        else {
            blue_maxr[i] = blue_maxr[i + 1] + 1;
            red_maxr[i] = -1;
        }
    }

    int ans = mmax(red_maxl[n - 1], blue_maxl[n - 1]) + mmax(red_maxr[0], blue_maxr[0]) + 2;

    for (int i = 1; i < n; ++i) {
        auto cur_ans = mmax(red_maxl[i - 1], blue_maxl[i - 1]) + mmax(red_maxr[i], blue_maxr[i]) + 2;
        ans = cur_ans > ans ? cur_ans : ans;
    }

    printf("%d", ans > n ? n : ans);

    return 0;
}