[codeforces 1328D] Carousel 分类讨论

Codeforces Round #629 (Div. 3)   比赛人数10967

[codeforces 1328D]  Carousel    分类讨论

总目录详见https://blog.csdn.net/mrcrack/article/details/103564004

也在线测评地址https://codeforces.ml/contest/1328/problem/D

Problem	Lang	Verdict	Time	Memory
D - Carousel	GNU C++11	Accepted	78 ms	1400 KB

思路同https://blog.csdn.net/HDUCheater/article/details/105133373

需要讨论的情况比较多，需要自己构造一些样例
容易观察得最多只需要三种颜色就可以了
当且仅当每个木马都相同时，k=1;
当k=2或3时，需要使用到贪心，
因为题目要求相邻的类型不同的颜色不同，
我们直接让所有的奇数项为1，偶数项为2，
唯一需要关注的便在于首项和末项是否冲突。

如果此时n为偶数，首项为1，末项为2，不冲突；

如果n为奇数，那就要让其中一对相邻的数涂同样的颜色，其他照旧12121212循环即可，
此时注意特判开头和结尾相同的情况；

如果没有任意一对相邻的数相同，
直接输出121212121212…3

AC代码如下

#include <cstdio>
#include <algorithm>
#define maxn 200010
using namespace std;
int a[maxn],b[maxn];
void solve(int same,int n){
	int i,x,tot,flag;
	if(same==1){//只有一种类型
		printf("1\n");
		for(i=0;i<n-1;i++)printf("1 ");
		printf("1\n");
	}else{//same>=2有二种或以上类型
		for(i=0;i<n;i++)
			b[i]=i%2+1;
		if(n%2==0){//偶数个n
			printf("2\n");
			for(i=0;i<n-1;i++)printf("%d ",b[i]);
			printf("%d\n",b[n-1]);
		}else{//奇数个n b[n-2]=2 b[n-1]=1 b[n]=1
			b[0]=0,tot=2,flag=0;
			for(i=1;i<n;i++)
				if(a[i]==a[i-1]&&!flag)b[i]=b[i-1],flag=1;//
				else b[i]=(b[i-1]+1)%2;
			b[n]=b[0];
			if(a[n-1]!=a[n]&&b[n-1]==b[n])b[n-1]=2,tot=3;
			printf("%d\n",tot);
			for(i=0;i<n-1;i++)printf("%d ",b[i]+1);
			printf("%d\n",b[n-1]+1);
		}
	}
}
int main(){
	int q,n,i,same;
	scanf("%d",&q);
	while(q--){
		same=1;
		scanf("%d",&n);
		scanf("%d",&a[0]);
		for(i=1;i<n;i++){
			scanf("%d",&a[i]);
			if(a[i]!=a[i-1])same=2;
		}
		a[n]=a[0];
		solve(same,n);
	}
	return 0;
}