#include "sphinx.h" #include <bits/stdc++.h> using namespace std; vector<int> find_colours(int n, vector<int> x1, vector<int> y) { vector<vector<int>> a(n, vector<int> ()); vector<int> e(n); int ox=0; vector<int> g(n, -1); int i=0; for(int i=0; i<n-1; i++) { a[x1[i]].push_back(y[i]); a[y[i]].push_back(x1[i]); } int s=0; for(int i=0; i<n; i++){if(a[i].size()==1){s=i; break;}} vector<int> h; vector<bool> vis(n, 0); g[s]=0; while(s!=-1) { h.push_back(s); vis[s]=1; int o=-1; for(auto x: a[s]){if(!vis[x])o=x;} s=o; } while(i<n-1) { e.assign(n, n); e[h[i]]=-1; e[h[i+1]]=-1; int u1=perform_experiment(e); if((n>2&&u1==2)||(n==2&&u1==1)){g[h[i+1]]=g[h[i]];} else {ox++; g[h[i+1]]=ox;}i++; } return g; }