HDU 5583 Kingdom of Black and White

接过队友的锅敲这道题，结果没太想清楚着急直接敲调试了好久。。又被队友嫌弃。。真心为我的代码能力担忧。。
“不要头脑一热就开始写”
下来检查发现当时的代码都写错了。。话说区域赛的数据为何这么弱。。

---

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=5583

题意：

给定01串，连续的相同字符的长度的记为距离。改变一个字符，使得最后的距离平方和最大。

分析：

对于每个元素记一个连续的前缀长度，一个连续的后缀长度，然后O(n)枚举一下就好了。
比赛的时候最后计算那里加加减减搞得很烦，实际化简一下就很好算了。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5, INF = 0x3f3f3f3f;
int a[maxn],f[2][maxn], f1[2][maxn];
char s[maxn];
bool flg[maxn];
int b[maxn];
int main (void)
{
    int t;scanf("%d",&t);
    for(int tt = 1; tt <= t; tt++){
        scanf("%s",s);
        int slen = strlen(s);
        ll lian = 0;
        ll yuan = 0;
        memset(flg, true, sizeof(flg));
        memset(f, 0 ,sizeof(f));
        memset(f1, 0 ,sizeof(f1));
        memset(b, 0 ,sizeof(b));
        for(int i = 0; i < slen; i++){
            a[i] = s[i]-'0';
        }
        for(int i = 0; i < slen; i++){
            if(i < slen - 1 && a[i] != a[i + 1]){
                flg[i] = false;
                flg[i +1] =false;
            }
            if(i == 0 || a[i] == a[i - 1]) lian++;
            else{
                yuan += lian * lian;
                lian = 1;
            }
        }
        yuan += lian * lian;
        for(int i=0;i<slen;i++){
            f[a[i - 1]][i] = b[a[i - 1]];
            b[!a[i]] = 0;
            b[a[i]]++;
        }
        memset(b, 0 ,sizeof(b));
        for(int i = slen - 1; i >= 0; i--){
            f1[a[i + 1]][i] = b[a[i + 1]];
            b[!a[i]] = 0;
            b[a[i]]++;
        }
        ll aans = yuan;
        ll res;
        for(int i= 0;i < slen; i++){
            if(flg[i])  continue;
            if(a[i - 1] != a[i + 1]){
                if(a[i] == a[i + 1] || i == slen - 1)
                    res = yuan + 2 * f[a[i - 1]][i]  - 2 * f1[a[i + 1]][i];
                else if(a[i] == a[i - 1] || i == 0) 
                    res = yuan - 2 * f[a[i - 1]][i] + 2 * f1[a[i + 1]][i];
            }else  res = yuan + 2 * f[a[i - 1]][i] + 2 * f1[a[i + 1]][i] + 2ll *  f[a[i - 1]][i]  * f1[a[i + 1]][i];
             aans = max(aans, res);
        }
        printf("Case #%d: %lld\n", tt, aans);
    }
    return 0;
}