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;
}