Contest1002 - HHU ACM 综合训练1 A题 Kingdom of Black and White（朴素）

来源：互联网发布：坏男孩软件靠谱吗编辑：程序博客网时间：2024/04/29 20:25

题意：给定一串01序列，序列的值表示为序列中连续的0或1的个数的平方相加的和。最多可以改变其中一个数字变为0或1，求序列最大值。

思路：很朴素的想法，一开始我以为有什么简便算法，然而并没有= =

首先离散化，数组存储的是连续的0或1的个数，之后对这个数组进行遍历，每次把a[i]+1的平方+a[i+1]-1的平方计算出来的和与a[i]的平方+a[i+1]的平方计算出来的和相比较，若更新后的值大，则更新最大差值。a[i]若为1则需要将a[i]左右相加求平方和并与之前的平方和比大小，依旧需要判断之后更新最大差值。

#include <iostream> #include <string.h> #include <algorithm> using namespace std; long long a[100005];   int main(){           int t;     //freopen("test.in","r",stdin) ;     cin>>t;     for(int i=1;i<=t;i++){         int cnt=0;         memset(a,0,sizeof(a));         string s;         cin>>s;         long long sum=0;                   for(int j=0;j<s.length();j++){             if(j==0)                 a[cnt]++;             else{                 if(s[j]==s[j-1])                     a[cnt]++;                 else{                     cnt++;                     a[cnt]=1;                 }             }         }                             for(int j=0;j<=cnt;j++)             sum+=a[j]*a[j];         //cout<<sum<<endl;         long long maxn=0;         for(int j=0;j<=cnt;j++){             if(a[j]==1){                 if(j==cnt){                     long long NEW = (a[j-1]+1)*(a[j-1]+1);                     long long OLD = a[j-1]*a[j-1]+1;                     if(NEW>OLD){                      maxn = max(maxn,NEW-OLD);                     }                 }                 else if(j==0){                     long long NEW = (a[j+1]+1)*(a[j+1]+1);                     long long OLD = 1+a[j+1]*a[j+1];                     if(NEW>OLD){                      maxn = max(maxn,NEW - OLD);                     }                 }                 else{                     long long NEW = (a[j-1]+a[j+1]+1)*(a[j-1]+a[j+1]+1);                     long long OLD = a[j-1]*a[j-1]+1+a[j+1]*a[j+1];                     if(NEW>OLD){                      maxn = max(maxn,NEW - OLD);                     }                 }               }         }         for(int j=0;j<cnt;j++){             long long NEW = (a[j]+1)*(a[j]+1)+(a[j+1]-1)*(a[j+1]-1);             long long OLD = a[j]*a[j]+(a[j+1])*(a[j+1]);             if(NEW>OLD){                 maxn=max(maxn,NEW-OLD);             }         }         sum+=maxn;         printf("Case #%d: ",i);           cout<<sum<<endl;     }      return 0; }

0 0