Codeforces Round #629 (Div. 3) D. Carousel（一个圈连续涂颜色）

链接

http://codeforces.com/problemset/problem/1328/D

题意

给一个圈动物涂颜色，连续不同动物必须图不同颜色，连续相同动物怎么图都可以，问最少几种颜色。

题解

1. 一种动物时，消耗1种颜色
2. n为偶数时，消耗2种颜色（1，2，1，2，1，2~）
3. n为奇数时，存在连续相同动物，消耗2种颜色，因为凡是存在连续相同动物，可以化成偶数情况 （第一个和最后一个也算是连续）
4. n为奇数时，不存在连续相同，还是1，2走，最后一个挂3。（1，2，1，2…3）

#include <iostream>
#include <unordered_set> 
using namespace std;
int n;
int a[200005];
int dp[200005];
unordered_set<int>sets;
void solve()
{
	int ans=0;
	if(sets.size()==1)
	{
		ans=1;
	}
	else if(n%2==0)
	{
		ans=2;
		for(int i=1;i<=n;i++)
		{
			if(i%2==1)
			dp[i]=1;
			else
			dp[i]=2;
		}
	}
	else if(n%2==1)
	{
		if(a[1]==a[n])
		{
			ans=2;
			for(int i=1;i<=n;i++)
			{
				if(i%2==1)
				dp[i]=1;
				else
				dp[i]=2;
			}
		}
		else
		{
			for(int i=2;i<=n;i++)
			{
				int top,flag=0;
				if(a[i]==a[i-1])
				{
					top=i;
					ans=2;
					for(int j=1;j<=n;j++)
					{
						if(j==top)
						{
							flag=1;
							dp[j]=dp[j-1];
						}
						else if(flag==0)
						{
							if(j%2==1)
							dp[j]=1;
							else
							dp[j]=2;
						}
						else if(flag==1)
						{
							if(j%2==0)
							dp[j]=1;
							else
							dp[j]=2;
						}
					}
					break;
				}
			}
		}
		
	}
	if(ans==1)
	{
		cout<<ans<<endl;
		for(int i=1;i<=n;i++)
		cout<<"1 ";
		cout<<endl;
	}
	else if(ans!=0)
	{
		cout<<ans<<endl;
		for(int i=1;i<=n;i++)
		cout<<dp[i]<<" ";
		cout<<endl;
	}
	else
	{
		cout<<"3"<<endl;
		for(int i=1;i<n;i++)
		{
			if(i%2==1)
			cout<<"1 ";
			else
			cout<<"2 ";
		}
		cout<<"3"<<endl;
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		sets.clear();
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sets.insert(a[i]);
		}
		solve();
	}
}