#include<bits/stdc++.h>
#include "mushrooms.h"
#define MAXN 20000
using namespace std;
int n,bucket,res,ans;
int color[MAXN],br[2];
vector<int> s1,s2,curr;
int count_mushrooms(int N){
n=N;
bucket=min(int(sqrt(n)*1.5)*n,n);
for(int i=1;i<bucket;i++){
color[i]=use_machine({0,i});
}
for(int i=0;i<bucket;i++){
br[color[i]]++;
if(color[i]==0)s1.push_back(i);
else s2.push_back(i);
}
if(br[0]>br[1]){
ans=br[0];
for(int i=bucket;i<n;i+=br[0]){
curr.clear();
for(int f=i;f<min(i+br[0],n);f++){
curr.push_back(s1[f-i]);
curr.push_back(f);
}
res=use_machine(curr);
ans+=br[0]-ceil(res/2.0);
}
}else{
ans=br[0];
for(int i=bucket;i<n;i+=br[1]){
curr.clear();
for(int f=i;f<min(i+br[1],n);f++){
curr.push_back(s2[f-i]);
curr.push_back(f);
}
res=use_machine(curr);
ans+=ceil(res/2.0);
}
}
return ans;
}
/*
int main(){
}
*/
