Решение задачи Ехаб и неПутевые MEXы с Codeforces

#include <bits/stdc++.h>

using namespace std;

int ans[100005];

int main() {
	memset(ans, -1, sizeof ans);
	int n;
	cin >>n;
	vector<int> v[n+1];
	
	for(int x, y, i = 1; i < n; i++) cin >>x >>y, v[x].push_back(i), v[y].push_back(i);

	for(int k = 3, i = 1; i <= n; i++) if(v[i].size() >= 3) {
		for(int j = 0; j < 3; j++) ans[v[i][j]] = j;
		for(int j = 1; j < n; j++) cout<<(ans[j] == -1 ? k++ : ans[j]) << "\n";
		exit(0);
	}
	
	for(int i = 0; i < n-1; i++) cout << i << "\n";
}