Codeforces 780C-Andryusha and Colored Balloons dfs

最新推荐文章于 2020-12-11 23:11:20 发布

原创最新推荐文章于 2020-12-11 23:11:20 发布 · 274 阅读

0 ·

CC 4.0 BY-SA版权

cf 同时被 2 个专栏收录

54 篇文章

订阅专栏

dfs

25 篇文章

订阅专栏

本文介绍了一种解决树形图染色问题的算法。该算法通过深度优先搜索（DFS）来为树上的每个节点分配颜色，确保任意三个相邻节点颜色不同。文中提供了一个完整的C++实现代码示例。

题意：给定一颗树，要在结点上涂色。规定要求每三个直接相连的结点颜色不能相同。求用最少的颜色使满足要求，并输出颜色结点颜色。

#include<cstdio>
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,ox3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
const int INF = ( 2e9 ) + 2;
const int maxn = 2e5+10;
vector<int> g[maxn];
int c[maxn],ans;
void dfs(int u,int fa)
{
	int x=0;							//避免兄弟结点颜色相同 
	for(int i=0,L=g[u].size();i<L;i++)
	{
		int v=g[u][i];
		if(v==fa||c[v])continue;		// 这个点已经染色 
		x++;
		while(x==c[u]||x==c[fa])x++;	//避免与父节点和祖先结点颜色相同 
		c[v]=x;
		dfs(v,u);
	}
	if(x>ans)ans=x;
}
int main()
{
	int n;
//	freopen("in.txt","r",stdin);
	scanf("%d",&n);
	for(int i=0;i<n-1;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	ans=-1;
	c[1]=1;
	dfs(1,0);
	printf("\n%d\n",ans);
	for(int i=1;i<=n;i++)
	printf("%d ",c[i]);
}