codeforces 1328D Carousel 构造+思维

题目大意：给出$n$个不同的动物，它们围成一个圈，$t[i]$表示第$i$个动物的种类，你要给这些动物染色，使得没有相邻的不同种类的动物被染成了相同的颜色，同一种类的动物可以被染成不同颜色。请输出最少需要的颜色数，并输出这$n$只动物的颜色。

思路：题面有毒，看了好久才看懂什么意思。构造题，首先特判掉动物种类都相同的情况，然后分奇偶，偶数一定可以按照$1、2、1、2\dots \dots$的方式构造，这样相邻的颜色均不相同$(b[n]、b[1]$也不同$)$；奇数呢我们也按照$1、2、1、2\dots \dots$的方式构造，如果$t[n]、t[1]$不同但是$b[n]、b[1]$相同，这时是不满足题意的，我们从$n$开始向前扫，找到第一个满足$t[i]=t[i-1]$的位置，令$b[i]=b[i-1]$，从$i+1$开始重新按照上面的方式构造(具体看代码)就可以了，为什么呢？因为这样相当于把$b[i\dots n]$与原来的构造方式翻转了一下，所以$b[n]、b[1]$肯定不同了，但是找不到这样的位置的话就必须把$t[n]$染成新的染色。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=2e5+5;

int a[maxn],b[maxn];

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        bool flag=1;
        for(int i=2;i<=n&&flag;i++)
            if(a[i]!=a[i-1])
                flag=0;
        if(flag)
        {
            printf("1\n");
            for(int i=1;i<=n;i++)
                printf("1 ");
            printf("\n");
            continue;
        }
        if(n%2==0)
        {
            printf("2\n");
            for(int i=1;i<=n;i++)
            {
                if(i&1)
                    printf("1 ");
                else
                    printf("2 ");
            }
            printf("\n");
        }
        else
        {
            int k=2;
            for(int i=1;i<=n;i++)
            {
                if(i&1)
                    b[i]=1;
                else
                    b[i]=2;
            }
            if(a[n]!=a[1]&&b[n]==b[1])
            {
                int idx=0;
                for(int i=n;i>=2;i--)
                {
                    if(a[i]==a[i-1])
                    {
                        idx=i;
                        break;
                    }
                }
                if(idx)
                {
                    b[idx]=b[idx-1];
                    for(int i=idx+1;i<=n;i++)
                    {
                        if(b[i-1]==2)
                            b[i]=1;
                        else
                            b[i]=2;
                    }
                }
                else
                    b[n]=++k;
            }
            printf("%d\n",k);
            for(int i=1;i<=n;i++)
                printf("%d ",b[i]);
            printf("\n");
        }
    }
    return 0;
}