#include "richest.h"
#include <bits/stdc++.h>

using namespace std;
typedef int ll;

ll on2(ll n) {
    vector<ll> a, b;
    for (ll i = 1; i < n; i++) {
        for (ll j = 0; j < i; j++) {
            a.push_back(i);
            b.push_back(j);
        }
    }

    vector<ll> c = ask(a, b);
    std::reverse(c.begin(), c.end());

    vector<vector<bool>> cmp(n, vector<bool>(n, true));
    for (ll i = 1; i < n; i++)
        for (ll j = 0; j < i; j++) {
            cmp[i][j] = c.back() == i;
            cmp[j][i] = c.back() == j;
            c.pop_back();
        }

    for (ll i = 0; i < n; i++) {
        bool good = true;
        for (ll j = 0; j < n; j++)
            good = good and cmp[i][j];

        if (good)
            return i;
    }
    return 0;
}

ll onlogn(ll n) {
    mt19937 rng(3498);

    vector<ll> p(n);
    std::iota(p.begin(), p.end(), 0);

    while (p.size() > 1) {
        std::shuffle(p.begin(), p.end(), rng);
        ll P = (ll)p.size();

        vector<ll> a(P), b(P);
        for (ll i = 0; i < P; i++) {
            a[i] = p[i];
            b[i] = p[(i + 1) % P];
        }

        a = ask(a, b);
        p.clear();
        for (ll i = 0; i < P; i++) {
            if (a[i] == a[(i + 1) % P]) {
                p.push_back(a[i]);
                i++;
            }
        }
    }

    return p[0];
}

ll richest(ll N, ll T, ll S) {
    if (T == 1) {
        return on2(N);
    } else {
        return onlogn(N);
    }
}