D. Carousel

题目链接

题目大意：一个旋转木马上有n只编号不同的动物座椅围成一圈(一种编号一种动物 ) ，现在给他涂色，要求颜色种类用最少且前后不同种动物的颜色不能相同

解题思路： 分类讨论：
一：n是偶数，则只需两种颜色按1，2重复涂下去即可
二：n是奇数：
1）若存在相邻的同种动物座椅，则两个座椅可以看作一个来涂色，按n是偶数来涂。
2）不存在相邻的同种动物座椅，则按1，2顺序涂下去，最后一个涂第三种即可
特殊情况：只有一种动物，则全部涂1即可。

#include<iostream>
#include<cstring>
#define maxn 1000010
int n,t[maxn],ans[maxn],s[maxn],count=0;
using namespace std;
int main()
{
	int q;
	cin>>q;
	while(q--)
	{
		count=0;
		cin>>n;
		memset(s,0,n);
		for(int i=0;i<n;i++)
		{
			cin>>t[i];
			if(!s[t[i]])
			{
				s[t[i]]=1;
				count++;//count用来记录动物种数
			}
		}
		if(count==1)
		{
			cout<<count<<endl;
			for(int i=0;i<n;i++)
			{
				cout<<1<<" ";
			}
			cout<<endl;continue;
		}//特殊情况
		if(n%2)//n为奇数
		{
			int flag=0;
			for(int i=0;i<n;i++)
			{
				if(t[i]==t[(i+1)%n])//存在相邻的同种动物座椅
				{
					flag=1;
					cout<<2<<endl;
					for(int pos=i+1; pos<n;++pos)
					{
						if((pos-1)%2) ans[pos]=1;//pos-1即把i和i+1看作一个座位
						else ans[pos]=2;
					}
					for (int pos=i;pos>=0;--pos)
					{
						if(pos%2) ans[pos]=1;
						else ans[pos]=2;
					}
					for(int j=0;j<n;++j) cout<<ans[j]<<" ";
					cout<<endl;break;
				}
			}
			if(flag) continue;
			else//二 2）
			{
				cout<<3<<endl;
				for(int i=0;i<n-1;i++)
				{
					if(i%2) cout<<"2 ";
					else cout<<"1 ";
				}
				cout<<"3 ";	
				cout<<endl;
			}
		}
		else//n为偶数
		{	
			cout<<2<<endl;
			for(int i=0;i<n;i++) 
			{
				if(i%2) cout<<"2 ";
				else cout<<"1 ";
			}
			cout<<endl;
			continue;
		}
	}
	return 0;
 }