AtCoder Beginner Contest 146 D.Coloring Edges on Tree

AtCoder Beginner Contest 146 D.Coloring Edges on Tree

题目链接

题意比较简单，就是用最少的颜色上色，使得连在一个点上的所有边颜色都不同
答案很好找，就是最大的度数，难点在于上色，可以这么考虑：

• 找到度数最大的点，以此点为起点进行 BFS，设一个点的颜色数组，起点的颜色置为 $0$，当以某个点 BFS 时，所有与之相连的点都不能与其同色
• 每次 BFS，都从 $1$ 颜色开始上色，若与起点颜色重复或者某点上色完毕颜色都要加 $1$ 取模
• 但是题目要求输出每条边的颜色，我们可以用 map+pair 记录每条边的颜色，不难发现边的颜色和 BFS 过程中某点上的颜色相同

AC代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
vector<int>G[N];
int vis[N],color[N],n,u,v,mx=0,s;
struct node{
    int u,v;
};
vector<node>edge;
map<pair<int,int>,int>m;
void bfs(int s){
    queue<int>q;
    q.push(s);
    vis[s]=1;
    while(!q.empty()){
        int u=q.front();
        q.pop();
        int c=1;
        for(auto v:G[u]){
            if(!vis[v]){
                q.push(v);
                if(c==color[u]) (c==mx-1)?c=mx:(c=(c+1)%mx);
                color[v]=c;
                m[{min(u,v),max(u,v)}]=c;
                (c==mx-1)?c=mx:(c=(c+1)%mx);
                vis[v]=1;
            }
        }
    }
}
int main(){
    cin>>n;
    for(int i=0;i<n-1;i++){
        cin>>u>>v;
        edge.push_back({u,v});
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for(int i=1;i<=n;i++){
        if(G[i].size()>mx){
            mx=G[i].size();
            s=i;
        }
    }
    bfs(s);
    color[s]=0;
    cout<<mx<<endl;
    for(auto i:edge) cout<<m[{min(i.u,i.v),max(i.u,i.v)}]<<endl;
    return 0;
}