HDU5583 上海赛铜牌题

　　这道题的意思是给你一个01串， 定义这串的优美值为连续相同的数字的平方和， 现在可以改变这些串中一个字符， 问你优美值最大是多少？ 我们可以预处理出d1[i]i左边和str[i]相同的字符个数， d2[i]i右边和str[i]相同的字符个数， 对于一个i， 我们可以进行如下分类， i周围的两个元素和i都不相同， 有一个相同， 两个都相同， 按照这个规律扫一遍即可得出答案， 代码如下：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
char str[100000 + 100];
int d1[100000+100], d2[100000 + 100];

LL n2(LL a)
{
return a*a;
}

int main()
{
int T;
scanf("%d", &T);
int kase = 0;
while(T--)
{
scanf("%s", str+1);
int len = strlen(str+1);
memset(d1, 0, sizeof(d1));
memset(d2, 0, sizeof(d2));
LL res = 0;
for(int i=1; i<=len; i++)
{
if(i>1 && str[i]!=str[i-1])
{
res += (LL)d1[i-1]*(LL)d1[i-1];
d1[i] = 1;
}
else d1[i] = d1[i-1]+1;
}
res += (LL)d1[len]*(LL)d1[len];
for(int i=len; i>=1; i--)
{
if(i<len && str[i]!=str[i+1])
d2[i] = 1;
else d2[i] = d2[i+1] + 1;
}
LL ans;
LL ansres = res;
str[0]='3', str[len+1]='3';
for(int i=1; i<=len; i++)
{
if(str[i]==str[i+1] && str[i]==str[i-1])
{
ans = res-n2(d1[i]+d2[i]-1)+n2(d1[i]-1)+n2(d2[i]-1)+1;
ansres = max(ansres, ans);
}
if(str[i]==str[i+1] && str[i]!=str[i-1])
{
ans = res-n2(d2[i])-n2(d1[i-1])+n2(d1[i-1]+1)+n2(d2[i]-1);
ansres = max(ansres, ans);
}
if(str[i]!=str[i+1] && str[i]==str[i-1])
{
ans = res-n2(d1[i])-n2(d2[i+1])+n2(d1[i]-1)+n2(d2[i+1]+1);
ansres = max(ansres, ans);
}
if(str[i]!=str[i-1] && str[i]!=str[i+1])
{
ans = res-n2(d1[i-1])-n2(d2[i+1])-1+n2(d1[i-1]+d2[i+1]+1);
ansres = max(ansres, ans);
}
}
printf("Case #%d: %lld\n", ++kase, ansres);
}
return 0;
}