Codeforces 782C Andryusha and Colored Balloons dfs

点击打开链接

题意:n<=2e5的tree 求任意三个相邻顶点都不同色,所需要的颜色种数和上色方案 

dfs遍历时,保证当前子结点颜色和祖先u,fa[u]结点颜色不同 && 兄弟结点的颜色要不同即可

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e5+20;
int n,c[N],ans;
pair<int,int> p[N];
vector<int> e[N];
void dfs(int u,int fa)
{
	int k=1;
	for(int i=0;i<e[u].size();i++)
	{
		int v=e[u][i];
		if(v!=fa)
		{
			//保证当前子结点颜色和祖先a,b结点颜色不同 && 兄弟结点的颜色要不同 
			
			while(k==c[u]||k==c[fa])//最多找两次O(1) 
				k++;
			c[v]=k;
			k++;//兄弟结点的颜色要不同
			dfs(v,u);		
		}
	}
}
int main()
{
	while(cin>>n)
	{
		int u,v;
		for(int i=1;i<=n;i++)
			e[i].clear();
		memset(c,-1,sizeof(c));
		
		for(int i=1;i<=n-1;i++)
		{
			scanf("%d%d",&p[i].first,&p[i].second);
			u=p[i].first;
			v=p[i].second;
			
			e[u].push_back(v);
			e[v].push_back(u);
		}
		ans=0;
		c[1]=1;
		dfs(1,0);
		for(int i=1;i<=n;i++)
			ans=max(c[i],ans);
		cout<<ans<<endl; 
		for(int i=1;i<=n;i++)
		{
			printf("%d",c[i]);
			if(i==n)
			printf("\n");
			else
			printf(" ");
		}
	}
	return 0;
}