Codeforces Round #629 (Div. 3) D - Carousel-优快云博客

题意：

思路：统计段的个数，每一段1、2、1、2...这样填。如果有奇数个段，就选某一个长度大于1的段中间变一下。

注意：需要特别判断首尾能不能连成一段。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int a[N],b[N];
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n;
        scanf("%d",&n);

        int seg=0,last=0,now,t=2;
        for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            if(a[i]!=last) ++seg;
            last=a[i];
        }

        if(last==a[1]) --seg;//如果首尾相同，是一段，段的个数减1
        last=0;
        bool mark=1;

        for(int i=1;i<=n;++i){
            if(a[i]==last){
                if(mark&&(seg&1)){//应该变
                    mark=0;//已变
                    t=3-t;
                }
            }else{
                t=3-t;
            }
            b[i]=t;
            last=a[i];
        }

        if(mark&&(seg&1)&&a[n]!=a[1]) b[n]=3; //应该变&&首尾不同 3
        else if(mark&&(seg&1)&&a[n]==a[1]) b[n]=2;//应该变&&首尾相同（首尾就是找到的第1个长度大于1的连续的段，就在这里变！）

        if(b[n]==3) puts("3");
        else if(seg>1) puts("2");
        else puts("1");
        
        for(int i=1;i<=n;++i) printf("%d ",b[i]);
        puts("");
    }
    return 0;
}