传送门
题意
有一个旋转木马,里面坐着n个动物,数字相同的表示相同的动物,不同的代表不同的动物,给这些动物染色,设定相邻的不同的动物颜色不同,相同的动物可以然相同颜色,求最少需要几种颜色。并输出染色情况。
思路
打比赛时没读懂题,早上补题的时候才看懂,就是一个构造+贪心思想,只有下面几种情况:
1.如果只有一种动物,则只需染一种色
2.如果为偶数,只需要两种色1212…1212,保证了结尾以开头也不一样
3.如果是奇数则又分为两种情况:
(1)、如果有相邻且相同的动物则可考虑最后一组相邻且相同动物一组染同一色,其他相邻染不同色,很好理解就是将最后一组合并为一个动物,然后个数又变成了偶数,染1212…12即可。
(2)、如果没有相邻且相同的动物,说明这个才是真正的奇数,则前面n-1(偶数)染1212…12最后染一个3即可。
AC代码
#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=2e5+5;
int arr[MAXN];
int main(){
int t;
cin>>t;
while(t--){
int n;
cin>>n;
for(int i=0;i<n;i++) cin>>arr[i];
bool flag=true;
int pos=-1;
for(int i=0;i<n;i++){
if(arr[i]!=arr[0]) flag=false;
if(arr[i]==arr[(i+1)%n]) pos=i;
}
if(flag){//全是相同的,都染1
puts("1");
for(int i=0;i<n-1;i++)
cout<<1<<" ";
puts("1");
}
else if(n%2==0){
puts("2");
for(int i=0;i<n-1;i++)
cout<<i%2+1<<" ";
puts("2");
}
else if(pos!=-1){
puts("2");
int k=0;
for(int i=0;i<n;i++){
cout<<(k+i)%2+1;
if(i==pos) k=1;
if(i<n-1) cout<<" ";
else cout<<endl;
}
}
else{
puts("3");
for(int i=0;i<n-1;i++)
cout<<i%2+1<<" ";
puts("3");
}
}
return 0;
}