#include "koala.h"
#include<bits/stdc++.h>
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int) x.size())
using namespace std;
int B[100]; int R[100];
int minValue(int N, int W) {
fill(B,B+N,0); B[0] = 1;
playRound(B,R);
for(int i = 0;i < N;i++){
if(R[i] == 0) return i;
}
}
int big[105];
int maxValue(int n, int W) {
fill(big,big+n,1);
vector<int> witness = {1,2,4,11};
for(int x : witness){
for(int i = 0;i < n;i++){
if(big[i]) B[i] = x;
else B[i] = 0;
}
playRound(B,R);
//for(int i = 0;i < n;i++) cout << R[i] << " ";
for(int i = 0;i < n;i++) if(R[i] == 0) big[i] = 0;
//cout << '\n';
}
for(int i = 0;i < n;i++) if(big[i]) return i;
return 0;
}
int greaterValue(int n, int W){
return -1;
}
void allValues1(int n, int W, int *P){
int B[n]; int R[n]; int A[n];
for(int i = 0;i < n;i++) A[i] = i;
stable_sort(A, A+n, [&](int a, int b){
fill(B,B+n,0);
B[a] = n, B[b] = n;
playRound(B,R);
if(R[a] == 0) return true;
else return false;
});
for(int i = 0;i < n;i++) P[A[i]] = i+1;
}
vector<int> precompX = {1,1,1,1,1,1,1,1,1,2,1,1,2,3,3,1,3,1,1,2,3,4,2,3,4,4,1,2,3,3,4,1,5,1,2,2,3,5,3,4,5,2,5,2,3,4,5,4,4,6,1,2,3,5,6,3,4,6,5,6,1,4,5,6,1,7,1,2,2,2,3,4,5,7,3,4,7,5,6,7,2,5,6,7,2,2,3,4,5,6,8,3,4,6,8,5,8,6,8};
int ptr = 0;
void allValues2(int n, int *P, int s, int e, vector<int> stuff){
if(sz(stuff) == 1){
P[stuff[0]] = s;
return;
}
int X = precompX[ptr]; ptr++;
fill(B,B+n,0);
for(int i : stuff) B[i] = X;
playRound(B,R);
vector<int> left, right;
for(int i : stuff){
if(R[i] == 0) left.push_back(i);
else right.push_back(i);
}
int m = s + sz(left);
allValues2(n, P, s, m-1, left);
allValues2(n, P, m, e, right);
}
void allValues(int N, int W, int *P) {
if (W == 2*N) allValues1(N, W, P);
else{
vector<int> stuff;
for(int i = 0;i < N;i++) stuff.push_back(i);
allValues2(N, P, 1, N, stuff);
}
}
