#include<bits/stdc++.h>
#include "mushrooms.h"
#define MAXN 20000
using namespace std;
int bucket,res,ans;
int color[MAXN],br[2];
vector<int> s1,s2,curr;
pair<int,int> p;
int count_mushrooms(int N){
bucket=min(int(sqrt(N)*1.414),N);
if(N==2){
return use_machine({0,1})+1;
}
bucket=min(int(sqrt(N)*1.5),N);
color[1]=use_machine({0,1});
color[2]=use_machine({0,2});
if(color[1]==0){
p={0,1};
}else if(color[2]==0){
p={0,2};
}else{
p={1,2};
}
for(int i=3;i<bucket;i+=2){
res=use_machine({p.first,i,p.second,i+1});
if(p.first==0){
if(res==0){color[i]=color[i+1]=0;}
if(res==1){color[i]=0; color[i+1]=1;}
if(res==2){color[i]=1; color[i+1]=0;}
if(res==3){color[i]=1; color[i+1]=1;}
}else{
if(res==0){color[i]=color[i+1]=1;}
if(res==1){color[i]=1; color[i+1]=0;}
if(res==2){color[i]=0; color[i+1]=1;}
if(res==3){color[i]=0; color[i+1]=0;}
}
}
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+=(min(i+br[0],N)-i)-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(){
}
*/
