题意:给你一个01字符串,要求最多改变一个字符(即0->1或1->0),使相邻相同字符的平方和最大
题解:其实此题还是暴力求解的,首先我们把相邻相同的字符分块(离散化),然后对于第i个分块与第i+1个分块,枚举(第i个分块字符个数+1和第i+1个分块字符个数-1)的情况与(第i个分块字符个数-1和第i+1个分块字符个数+1)的情况,需注意的是当第i+1个分块若只有一个字符,那么第i个分块、第i+1个分块、第i+2个分块会合成一个分块,同理,当第i个分块只有一个字符时,第i-1个分块、第i个分块、第i+1个分块会合成一个分块,我们更新最大值即可。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
char str[maxn];
ll sz[maxn];
int main()
{
int t,i,j;
ll k,ans,tmp;
scanf("%d",&t);
for(int ca=1;ca<=t;ca++)
{
scanf("%s",str);
for(i=1,j=1,k=1,tmp=0;str[i]!='\0';i++)
{
if(str[i]!=str[i-1])
{
sz[j++] = k;
tmp += k*k;
k=1;
}
else k++;
}
sz[j++]=k,tmp+=k*k,ans=tmp;
sz[j]=0;
for(i=1;i<j-1;i++)
{
if(sz[i+1]==1)
ans = max(ans,tmp-sz[i]*sz[i]-sz[i+1]*sz[i+1]-sz[i+2]*sz[i+2]+(sz[i]+sz[i+1]+sz[i+2])*(sz[i]+sz[i+1]+sz[i+2]));
else
ans = max(ans,tmp-sz[i]*sz[i]-sz[i+1]*sz[i+1]+(sz[i]+1)*(sz[i]+1)+(sz[i+1]-1)*(sz[i+1]-1));
if(sz[i]==1)
ans = max(ans,tmp-sz[i-1]*sz[i-1]-sz[i]*sz[i]-sz[i+1]*sz[i+1]+(sz[i-1]+sz[i]+sz[i+1])*(sz[i-1]+sz[i]+sz[i+1]));
else
ans = max(ans,tmp-sz[i]*sz[i]-sz[i+1]*sz[i+1]+(sz[i]-1)*(sz[i]-1)+(sz[i+1]+1)*(sz[i+1]+1));
}
printf("Case #%d: %lld\n",ca,ans);
}
return 0;
}