Codeforces Round #135 (Div. 2)-C. Color Stripe

Codeforces Round #135 (Div. 2)-C. Color Stripe

链接：http://codeforces.com/group/xrTA2IaQje/contest/249069/problem/E

思路：

1：m>2时分情况（与前后进行比较），m<=2时讨论ABAB和BABA，注意500005.。。。
m=1时不用变，但是我代码中没考虑到这个。
写的很杂乱，比较菜，希望大家可以指点和改正，谢谢。

#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
#include<set>
#include<map>
using namespace std;
int n,m;
char aa[500005];
int num=0;
char bb[4]= {' ','A','B','C'};
int main()
{
    cin>>n>>m;
    for(int i=1; i<=n; i++)
    {
        cin>>aa[i];
    }
    if(m>2)
    {
        for(int i=1; i<=n; i++)
        {
            if(aa[i]==aa[i-1])
            {
                // cout<<aa[i]<<endl;
                for(int j=1; j<=m; j++)
                {
                    if(aa[i]!=bb[j])
                    {
                        if(i==n)
                        {
                            if((bb[j]!=aa[i-1]))
                            {
                                aa[i]=bb[j];
                                // cout<<" hhhhh: "<<bb[j]<<endl;
                                num++;
                                break;
                            }
                        }
                        else
                        {

                            if((bb[j]!=aa[i+1])&&(bb[j]!=aa[i-1]))
                            {
                                aa[i]=bb[j];
                                num++;
                                break;
                            }
                        }


                    }
                }
            }
        }
   }
    else
    {
        int num1=0,num2=0;
        for(int i=1;i<=n;i++)
        {
            if(i%2==0)
            {
                if(aa[i]!='A')
                    num1++;
                if(aa[i]!='B')
                    num2++;
            }
            else
            {
                if(aa[i]!='B')
                    num1++;
                if(aa[i]!='A')
                    num2++;
            }
        }
        num=min(num1,num2);
        cout<<num<<endl;
        if(num1<=num2)
        {
            for(int i=1;i<=n;i++)
            {
                if(i%2==0)
                {
                    cout<<'A';
                }
                else
                {
                    cout<<'B';
                }
            }
            cout<<endl;
            return 0;
        }
        else
        {
            for(int i=1;i<=n;i++)
            {
                if(i%2==0)
                {
                    cout<<'B';
                }
                else
                {
                    cout<<'A';
                }
            }
            cout<<endl;
            return 0;
        }
    }
    cout<<num<<endl;
    for(int i=1; i<=n; i++)
    {
        cout<<aa[i];
    }
    cout<<endl;
}