#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
const int B = 137;
int count_mushrooms(int n)
{
vector<int> v[2];
v[0] = {0};
int i = 1;
for (; i < n && max(v[0].size(), v[1].size()) < B + 1; i += 2)
{
if (i + 1 < n)
{
int x = use_machine({0, i, i + 1});
if (x == 0) v[0].push_back(i), v[0].push_back(i + 1);
else if (x == 2) v[1].push_back(i), v[0].push_back(i + 1);
else
{
v[1].push_back(i + 1);
if (!use_machine({0, i})) v[0].push_back(i);
else v[1].push_back(i);
}
}
else v[use_machine({0, i})].push_back(i);
}
int res = 0;
for (int j = i; j < n; j += B)
{
vector<int> q;
int sz = min(n, j + B) - j;
for (int k = j; k < min(n, j + B); k++)
{
if (v[0].size() > v[1].size())
{
q.push_back(v[0][k - j]);
q.push_back(k);
}
else
{
q.push_back(v[1][k - j]);
q.push_back(k);
}
}
if (v[0].size() > v[1].size())
{
q.push_back(v[0].back());
res += sz - (use_machine(q) / 2);
}
else
{
q.push_back(v[1].back());
res += use_machine(q) / 2;
}
}
return res + v[0].size();
}
