传送门:
http://acm.hdu.edu.cn/showproblem.php?pid=5583
这是本年度上海区域赛的第二水题,也是mark一下我本年度的赛程,虽然最后打铁了!!!!
题意:
相当于给你一堆数,你最多可以将其中一个位置的数取反,使得相同连续的数的平方和最大!!!
就是把最大的数加1,相邻的数减1,暴力枚举一下位置,再注意特判一下1就ok了,因为它会将前后都聚到一起!
这种lowb题目样例就已经给出所有可能的情况了,写的时候注意一下,还有可能不变,所以最大值赋初值的时候一定要注意!
code:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int t,n;char s[maxn];ll a[maxn];
int main(){
cin>>t;int kk=1;
while(t--){
ll ans=0;
scanf("%s",s+1);memset(a,0,sizeof(a));
int len=strlen(s+1);ll sum=1,count=1;
ll maxx=0;
for(int i=2;i<=len;i++){
if(s[i]==s[i-1]){
sum++;
}
else{
maxx=max(maxx,sum);
ans=ans+sum*sum;
a[count++]=sum;sum=1;
}
}
a[count++]=sum;ans=ans+sum*sum;
ll tmp=0;ll anss=ans,tmpp=0;
for(int i=1;i<count;i++){
if(a[i]==maxx){
if(i==1)tmp=a[i+1];
else if(i==count-1) tmp=a[i-1];
else tmp=min(a[i-1],a[i+1]);
tmpp=ans+(ll)((a[i]+1)*(a[i]+1))-(ll)(a[i]*a[i]);
tmpp=tmpp+(ll)((tmp-1)*(tmp-1))-(ll)(tmp*tmp);
anss=max(anss,tmpp);
}
}
for(int i=1;i<count;i++){
if(a[i]==1){
tmpp=ans+(ll)((a[i]+a[i-1]+a[i+1])*(a[i]+a[i-1]+a[i+1]));
tmpp=tmpp-(ll)(a[i]*a[i])-(ll)(a[i-1]*a[i-1])-(ll)(a[i+1]*a[i+1]);
anss=max(tmpp,anss);
}
}
printf("Case #%d: %lld\n",kk++,anss);
}
}