CodeForces 1141G Privatization of Roads in Treeland (贪心+DFS染色)

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5+5;
#define pb push_back
int n,k;
struct Edge{int s,e;}e[N];
struct Edg {int to,loc;};
int deg[N],col[N],mark[N];
int colnum;
vector<Edg>G[N];

bool mycmp(int a,int b){ return a>b; }

void dfs(int u){     //在DFS过程中，贪心地对所有节点周围的边染色
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to;
        int id=G[u][i].loc;
        if(col[id])continue; 
        col[id]=mark[u]+1;mark[u]++;     //贪心地将u节点连向子节点的边全部优先染成不同颜色
        if(mark[u]>=colnum)mark[u]=0;     //如果出现冲突的话，就将这个点的其它连接子节点的边随便染成一种颜色
        mark[v]=col[id];if(mark[v]>=colnum)mark[v]=0;      //注意将下一个点也要做标记，这样做标记的目地就是尽可能的让一个节点周围的所有边不重复染色
        dfs(v);
    }
}

int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++){
        int u,v;scanf("%d %d",&u,&v);
        deg[u]++,deg[v]++;
        e[i]=Edge{u,v};
        G[u].pb(Edg{v,i});
        G[v].pb(Edg{u,i});
    }
    sort(deg+1,deg+1+n,mycmp);
    colnum=deg[k+1];        //得到第k+1个节点的度
    printf("%d\n",colnum);
    dfs(1);
    for(int i=1;i<n;i++)printf("%d ",col[i]);
}