每日Codeforces题解 [20210114]_c. andryusha and colored balloons-CSDN博客

本文链接：https://blog.csdn.net/qq_45530271/article/details/112632839

这篇博客讲述了如何解决Codeforces竞赛中的一道题目，涉及树状结构和颜色分配问题。通过深度优先搜索策略，找到使得树中所有节点颜色互不相同的最小颜色数，关键在于避免与父节点和祖父节点颜色重复。作者提供了详细的AC代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

- C题.[Codeforces 782C] Andryusha and Colored Balloons

C题.[Codeforces 782C] Andryusha and Colored Balloons

题目链接
思路： 给一棵树， $a - b - c$ 相连的三个节点颜色互不相同，所以我们选一个节点作为根，那么对于节点 $i$ 只需要保证：
①节点 $i$ 和它的兄弟不同色
②节点 $i$ 和它的父亲不同色
③节点 $i$ 和它的祖父（父亲的父亲）不同色
所以具体过程可以用dfs来实现，传递参数父亲节点号、祖父节点号、当前使用到的颜色计数器（这个值用引用传递比较方便）。每次开始向下遍历一个节点的所有儿子时，让这个颜色计数器从 $1$ 开始并每次递增，其中，若与父亲或者祖父颜色相同，就再往后选下一个颜色。同时更新使用的最大颜色编号。

//AC代码
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+7;
int fa[N],n,k=2,color[N];
vector<int>son[N];
void dfs(int now,int fa,int gra,int &c){
	while(c==color[fa]||c==color[gra])c++;
	color[now]=c;
	k=max(k,color[now]);
	if(son[now].size()==1&&now!=1)return;
	int cnt=0;
	for(auto i:son[now])
		if(i!=fa){
			cnt++;
			dfs(i,now,fa,cnt);
		}
}
int main(){
	cin>>n;
	for(int i=1,u,v;i<n;i++){
		scanf("%d%d",&u,&v);
		son[u].push_back(v);
		son[v].push_back(u);
	}
	int cnt=1;
	dfs(1,0,0,cnt);
	printf("%d\n",k);
	for(int i=1;i<=n;i++)printf("%d ",color[i]);
}