HDU-5583-暴力

https://vjudge.net/contest/167609#problem/K
给定一个串，规定sum为连续数字相同的长度的平方和。
比如 000011
sum为 4*4+2*2=20；
最多可以改变一个数字，也可以不改变，问你最大的sum是多少。
我的想法是改变的地方为 10 和01 ，总共有四种情况可以改变
10 变 11 和 00 01 变 00 和11 . 然后枚举四种情况。（寻找子串）
然而超时。
用暴力匹配，枚举合并的情况更好。但是要注意，一开始就要把总和记录起来，每次和总和比较，因为只让改变一次。

#include<iostream>
#include<cstring>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn = 60 +5;
ll sum[100006];
char a[100006];
int main ()
{   int t;
    scanf("%d",&t);
    for(int tt=1;tt<=t;tt++){
        memset(sum,0,sizeof(sum));
        scanf("%s",a+1);
        a[0]=a[1];
        int len=strlen(a+1);
        int j=1;
        for(int i=1;i<=len;i++){
            if(a[i]!=a[i-1]){
                j++;
                sum[j]=1;
            }
            else
            sum[j]++;
        }
        ll ans=0;
        sum[0]=0;
        for(int i=1;i<=j;i++){
            ans+=sum[i]*sum[i];
        }
        ll all=ans;

        sum[j+1]=0;
        for(int i=1;i<j;i++){
            if(sum[i]==1)
      {all=max(all,ans-sum[i-1]*sum[i-1]-sum[i]*sum[i]-sum[i+1]*sum[i+1]+(sum[i-1]+sum[i]+sum[i+1])*(sum[i-1]+sum[i]+sum[i+1]));

      }
               else
                {all=max(all,ans-sum[i+1]*sum[i+1]-sum[i]*sum[i]+(sum[i]-1)*(sum[i]-1)+(sum[i+1]+1)*(sum[i+1]+1));
                }

             if(sum[i+1]==1)
{all=max(all,ans-sum[i+2]*sum[i+2]-sum[i]*sum[i]-sum[i+1]*sum[i+1]+(sum[i+2]+sum[i]+sum[i+1])*(sum[i+2]+sum[i]+sum[i+1]));
}
               else
{all=max(all,ans-sum[i+1]*sum[i+1]-sum[i]*sum[i]+(sum[i+1]-1)*(sum[i+1]-1)+(sum[i]+1)*(sum[i]+1));
}


        }
        printf("Case #%d: %lld\n",tt,all);
    }
return 0;
}