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

#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;
}