CF 1025C Plasticine zebra

昨晚忘记判只有一个字符的情况fst了呜呜呜

挺有趣的题，昨晚连刚带猜弄出结论

考虑答案的取值，最优答案可能是一个后缀，或者是一个前缀，或者是一个后缀加上前缀

那么翻转之后最优答案的可选值就有了1的前缀+n的后缀

对于一个合法的zebra串，无论怎么翻转都不会影响他的取值，所以预处理一遍扫过去找答案即可。

强烈谴责CF昨晚的数学大赛pretest数据太水

Code：

#include <cstdio>
#include <cstring>
using namespace std;

const int N = 1e5 + 5;

int n, a[N], suml[N], sumr[N];
char s[N];

inline int max(int x, int y) {
    return x > y ? x : y;
}

inline int min(int x, int y) {
    return x > y ? y : x;
}

int main() {
    scanf("%s", s + 1);
    n = strlen(s + 1);
    for(int i = 1; i <= n; i++) a[i] = s[i] - '0';
    
    for(int i = 1; i <= n; i++) 
        if(a[i] != a[i - 1]) sumr[i] = sumr[i - 1] + 1;
        else sumr[i] = 1;
    for(int i = n; i >= 1; i--)
        if(a[i] != a[i + 1]) suml[i] = suml[i + 1] + 1;
        else suml[i] = 1;
    
    int ans = 0;
    for(int i = 1; i < n; i++) {
        if(a[i] != a[i + 1]) 
            ans = max(sumr[i] + suml[i + 1], ans);
        if(a[1] != a[n]) 
            ans = max(min(sumr[n], n - i) + min(suml[1], i), ans);
        ans = max(ans, suml[i]);
        ans = max(ans, sumr[i + 1]);
    }
    
    printf("%d
", max(ans, 1));
    return 0;
}