CF780C Andryusha and Colored Balloons

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
const int N=200005;
int n,tot,ans;
int d[N],head[N],Next[N*2],ver[N*2],c[N];
inline int read()
{
    int x=0;
    char c=getchar();
    bool flag=0;
    while(c<'0'||c>'9'){if(c=='-')flag=1;   c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<3)+(x<<1)+c-'0';c=getchar();}
    return flag?-x:x;
}
void add(int x,int y)
{
   ++tot;Next[tot]=head[x]; head[x]=tot; ver[tot]=y;
}//链式前向星建图
void dfs(int x,int fa)//x表示当前节点，fa表示父亲节点
{
    int now=0;
    for(int i=head[x];i;i=Next[i])
    {
        if(ver[i]!=fa)//把所以相邻的边涂色
        {
            ++now;
            while(now==c[fa]||now==c[x]) ++now;
            c[ver[i]]=now;
        }
    }
    for(int i=head[x];i;i=Next[i])
    if(ver[i]!=fa) dfs(ver[i],x);//一种分层的思想？？？
}
int main()
{
    n=read();
    for(int i=1;i<n;++i)
    {
        int x=read(),y=read();
        add(x,y),add(y,x);
        ++d[x],++d[y];
        ans=max(ans,d[x]),ans=max(ans,d[y]);//因为连接边数最大的边的儿子也必定最多，而一个节点的儿子互不相同，所以每个儿子都要一种颜色，最后加上自己颜色
    }
    cout<<ans+1<<endl;
    c[1]=1;
    dfs(1,0);
    for (int i=1;i<=n;++i) printf("%d%c",c[i]," \n"[i==n]);//最后那个是来自老师的特判写法
}

借鉴于YWHS官方刷题网站，思想来自于HJF老师