Codeforces Round #628 (Div. 2) C.Ehab and Path-etic MEXs（图论，思维题）

Posted 2021-03-05 kanoon

题意：

给一条树的 n-1 条边从 0 到 n-2 编号，使得任意两点路径中未出现的最小数最小的方案。

思路：

先给所有度为1的点所在边编号，之后其他点可以随意编排。

#include <bits/stdc++.h>
using namespace std;
 
const int M=110000;
 
vector<vector<int>> e(M);
vector<pair<int,int>> a;
 
map<pair<int,int>,int> m;
 
int main()
{
    int n;cin>>n;
    for(int i=0;i<n-1;i++){
        int u,v;cin>>u>>v;
        e[u].push_back(v);
        e[v].push_back(u);
        a.push_back(make_pair(u,v));
    }

    int num=1;
    for(int i=1;i<=n;i++){
        if(e[i].size()==1){
            for(int j:e[i]){
                pair<int,int> p1=make_pair(i,j);
                pair<int,int> p2=make_pair(j,i);
                if(!m[p1])
                    m[p1]=m[p2]=num++;
            }
        }
    }

    for(int i=1;i<=n;i++){
        if(e[i].size()!=1){
            for(int j:e[i]){
                pair<int,int> p1=make_pair(i,j);
                pair<int,int> p2=make_pair(j,i);
                if(!m[p1])
                    m[p1]=m[p2]=num++;
            }
        }
    }

    for(auto i:a){
        cout<<m[i]-1<<endl;
    }

    return 0;
}