hdu 5583 Kingdom of Black and White 2015ACM/ICPC亚洲区上海站-重现赛

先缩点，将颜色相同的连续line作为一个点。如果连通块只有一个frog，改变颜色后则会与相邻的两个连通块合为一体，改变量是(a+1+b)^2-(a^2+1+b^2)。如果连通块有多个frog，改变中间的颜色不会更优，因为是从(a+1+b)^2变为a^2+1+b^2只会更小。所以应该改变最左边或最右边frog的颜色，from a^2+b^2 to (a+1)^2+(b-1)^2 or (a-1)^2+(b+1)^2。

#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<string>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<cstring>
#include<vector>
#include<queue>
#include<map>

using namespace std;
//hdu 5583
const int maxn=1e5+10;
const int INF=0x3f3f3f3f;
int T;
long long N;
long long ans;
char seq[maxn];
long long presum;
class frog
{
public:
    long long num;
    int color;
public:
    frog()
    {
        num=0;
        color=0;
    }
    frog(int n,int c)
    {
        num=n;
        color=c;
    }
};
frog line[maxn];
int main()
{
    freopen("input.txt","r",stdin);
    scanf("%d",&T);
    for(int ca=1;ca<=T;ca++)
    {
        ans=0;
        memset(seq,0,sizeof(seq));
        memset(line,0,sizeof(line));
        scanf("%s",&seq);
        presum=0;
        N=strlen(seq);
        seq[N]='2';
        //cout<<seq<<endl;
        if(N==1)
        {
            printf("Case #%d: 1\n",ca);
            continue;
        }
        long long cnt=1;
        int idx=0;
        for(int i=0;i<N;i++)
        {
            if(seq[i]==seq[i+1])
            {
                cnt++;
//                if(i==N-2)
//                {
//                    line[idx]=frog(cnt,seq[i]-'0');
//                }
            }
            else
            {
                line[idx]=frog(cnt,seq[i]-'0');
                cnt=1;
                idx++;
//                if(i==N-2)
//                {
//                    line[idx]=frog(cnt,seq[i]-'0');
//                }
            }
        }
//        line[idx].num=1;
//        line[idx].color=seq[0]-'0';
//        for(int i=1;i<N;i++)
//        {
//
//            if(seq[i]==seq[i-1])
//            {
//                line[idx].num++;
//            }
//            else
//            {
//                idx++;
//                line[idx].num++;
//                line[idx].color=seq[i]-'0';
//            }
//        }
//        idx++;
        //cout<<"idx "<<idx<<endl;
        if(idx==1)
        {
            printf("Case #%d: %I64d\n",ca,N*N);//line[0].num*line[0].num);
            continue;
        }

        //for(int i=0;i<idx;i++) cout<<line[i].num<<" "<<line[i].color<<endl;
        //int maxlen=0;
        for(int i=0;i<idx;i++)
        {
            //cout<<"num^2 "<<(line[i].num^2)<<endl;
            presum+=line[i].num*line[i].num;
            //maxlen=max(maxlen,line[i].num);
//            if(maxlen<line[i].num)
//            {
//                maxlen=line[i].num;
//                maxidx=i;
//            }
            if(line[i].num==1)
            {
                if(i==0)
                {
                    ans=max(ans,(2*line[i+1].num));
                }
                else if(i==idx-1)
                {
                    ans=max(ans,(2*line[i-1].num));
                }
                else
                {
//                    if(line[i-1].color==line[i+1].color)
//                    {
                        ans=max(ans,2*(line[i-1].num+line[i+1].num+line[i-1].num*line[i+1].num));
                    //}
//                    else
//                    {
//                        ans=max(ans,(1+max(line[i-1].num,line[i+1].num)^2); must be 3 different colors
//                    }

                }

            }
        }
//        for(int i=1;i<idx;i++)
//        {
//            if(line[i].num==1)
//            {
//                ans=max(ans,2*(line[i-1].num+line[i+1].num+line[i-1].num*line[i+1].num));
//            }
//        }
//        for(int i=0;i<idx;i++)
//        {
//            if(line[i].num==maxlen)
//            {
//                if(i==0) ans=max(ans,2*(line[i].num-line[i+1].num)+2);
//                else if(i==idx-1) ans=max(ans,2*(line[i].num-line[i-1].num)+2);
//                else
//                {
//                    int b=min(line[i-1].num,line[i+1].num);
//                    ans=max(ans,2*(line[i].num-b)+2);
//                }
//            }
//        }
        for(int i=0;i<idx-1;i++)//current+1, right-1
        {
//            if(line[i].num>=line[i+1].num)
//            {
                ans=max(ans,2*(line[i].num-line[i+1].num)+2);
//            }
//            else
//            {
//                ans=max(ans,2*(line[i+1].num-line[i].num)+2);
//            }
        }
        for(int i=1;i<idx;i++)//current+1, left-1
        {
            ans=max(ans,2*(line[i].num-line[i-1].num)+2);
        }
        //cout<<presum<<endl;
        printf("Case #%d: %I64d\n",ca,presum+ans);
    }
}