#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
//#define int long long
#define pb push_back
#define fi first
#define se second
#define dd(x) cout<<#x<<" is "<<x<<endl;
#define dd2(x,y) cout<<#x<<" is "<<x<<" "<<#y<<" is "<<y<<endl;
#define dl(x) cout<<#x<<" is "<<endl; for(auto i:x) cout<<i<<" "; cout<<endl;
vector<int>a;
vector<int>b;
int res;
void query(vector<int> &v, vector<int>&x, int n){
vector<int>m;
for(int i=0; i<n; i++) {
m.pb(v[i]);
m.pb(x[i]);
}
res += n-(use_machine(m)+1)/2;
}
int count_mushrooms(int n) {
a.pb(0);
for(int i=1; i<n; i++){
if(!use_machine({0,i})) a.pb(i);
else b.pb(i);
if(a.size() == 100 or b.size() == 100) break;
}
bool f=0;
if(b.size() > a.size()) swap(a,b), f=1;
int st=a.size() + b.size();
vector<int>test;
int sz = a.size();
for(int i=st; i<n; i++){
test.pb(i);
if(test.size()==sz){
query(a,test,sz);
test.clear();
}
}
if(test.size()) query(a,test,test.size());
res += a.size();
if(f) res=n-res;
return res;
}
