Codeforces Round #547 (Div. 3) G. Privatization of Roads in Treeland

题目链接： G. Privatization of Roads in Treeland

题目大意：

 给你一个图，点代表一个城市，边代表路，一个城市的路不能有两条或两条以上属于同一个公司，如果有则认为
 这个城市是不好（not good）的；现在给出一个n图，n-1条边，允许要k个不好的点，问你至少需要多少个公司才
 能满足条件，输出，并打印这n-1天边属于哪一个公司。

解题思路：这是一个染色题，先对度排序，找到第k+1大的数ans就是答案了，在dfs一边，用ans种
颜色染色就好了。这里我被怎么不重复染色卡了一下，我看了题解的处理方法，觉得非常简单和巧
妙，就是在dfs加上一个参数color表示当前进入该点的边的颜色，上颜色的时候模一下就好了。
简单证明一下：因为dfs是对于每一个点和边都是访问一遍，所以从当前颜色加一取模既不会重复，
也不会用到超过ans中颜色。又get到一种简单有效的方法了，嘻嘻

ac代码：

//#include <iostream>
#include <bits/stdc++.h>

using namespace std;

const int maxn=2e5+5;
#define ll long long
const int inf =2e5+1000;
map<pair<int,int>,int>edge;
vector<int>u[maxn];
pair<int,int>uu[maxn];
bool vis[maxn];
int du[maxn];
bool cmp(int a,int b){
    return a>b;
}
//color是颜色，k是允许使用的颜色数
void dfs(int rt,int k,int color){
    vis[rt]=1;
    //cout<<rt<<endl;
    int len=u[rt].size();
    for(int i=0; i<len; i++){
        int v=u[rt][i];
        if(!vis[v]) {
            color=edge[{rt,v}]=edge[{v,rt}]=color%k+1;//这是一个上色的过程
            //这里是先模再加还是先加再模要视情况而定
            dfs(v,k,color);
        }
    }

}
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=0; i<n+5; i++) du[i]=0,vis[i]=0;
    for(int i=0; i<n-1; i++){
        int x,y;
        scanf("%d%d",&x,&y);
        u[x].push_back(y);
        u[y].push_back(x);
        du[x]++,du[y]++;
        uu[i]={x,y};
    }
    sort(du+1,du+n+1,cmp);
    int ans=du[k+1];
    dfs(1,ans,0);
    //cout<<edge[{5,2}]<<" "<<edge[{2,5}]<<endl;
    cout<<ans<<endl;
    for(int i=0; i<n-1; i++){
        printf("%d ",edge[uu[i]]);
    }
    return 0;
}