#include "sphinx.h"
#include <bits/stdc++.h>
using namespace std;
#define vi vector<int>
#define pb push_back
#define pii pair<int,int>
#define ff first
#define ss second
#define all(x) x.begin(),x.end()
#define ll long long int
const int N = 250;
vi g[N];
int ask(vi E){
return perform_experiment(E);
}
std::vector<int> find_colours(int n, std::vector<int> X, std::vector<int> Y) {
for(int i = 0; i < n; ++i) g[i].clear();
for(int i = 0; i < n - 1; ++i) g[X[i]].pb(Y[i]), g[Y[i]].pb(X[i]);
vi col(n);
for(int i = 0; i < n; ++i){
for(int j = 0; j < n; ++j){
vi E(n, j);
E[i] = -1;
if(ask(E) == 1){
col[i] = j;
break;
}
}
}
return col;
}
