#include "sphinx.h"
#include <bits/stdc++.h>
using namespace std;
int n, m;
vector<vector<int>> g;
const int MN = 251;
int par[MN], sz[MN];
int find(int u) {
    if (par[u] == u) return u;
    return par[u] = find(par[u]);
}
bool merge(int u, int v) {
    u = find(u);
    v = find(v);
    if (u == v) return false;
    if (sz[u] > sz[v]) swap(u, v);
    par[u] = v;
    sz[v] += sz[u];
    return true;
}
vector<int> find_colours(int N, vector<int> X, vector<int> Y) {
    n = N;
    m = Y.size();
    g.resize(n);
    for (int i = 0; i < m; i++) {
        g[X[i]].push_back(Y[i]);
        g[Y[i]].push_back(X[i]);
    }
    for (int i = 0; i < n; i++) {
        par[i] = i;
        sz[i] = 1;
    }
    vector<int> willAsk(n, n);
    for (int i = 1; i < n; i++) {
        willAsk[i] = willAsk[i-1] = -1;
        int tot = perform_experiment(willAsk);
        int should = 2;
        if (i > 1) should++;
        if (i < n-1) should++;
        if (tot != should) {
            merge(i - 1, i);
        }
        willAsk[i] = willAsk[i-1] = n;
    }
    vector<int> ans(n);
    ans[0] = 0;
    for (int i = 1; i < n; i++) {
        if (find(i) == find(i-1)) ans[i] = ans[i-1];
        else ans[i] = ans[i-1] + 1;
    }
    return ans;
}
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict  | Execution time | Memory | Grader output | 
|---|
| Fetching results... |