#include <bits/stdc++.h>
#include "mushrooms.h"
using namespace std;
int count_mushrooms(int n) {
vector<int>mac;
int c1 = use_machine(mac = {0, 1});
if(n==2) return 1 + 1-c1;
int c2 = use_machine(mac = {0, 2});
int tp[n];
tp[0] = 1;
tp[1] = 1-c1;
tp[2] = 1-c2;
int s1, s2;
if(c1 && c2) s1 = 1, s2 = 2;
else {
s1 = 0;
if(c1) s2 = 2;
else s2 = 1;
}
for(int i=3; i<n; i+=2) {
if(i==n-1) {
tp[i] = 1 - use_machine(mac = {0, i});
} else {
int c = use_machine(mac = {i, s1, i+1, s2});
if(c==0) {
tp[i] = tp[i+1] = tp[s1];
} else if(c==1) {
tp[i] = 1-tp[s1];
tp[i+1] = tp[s1];
} else if(c==2) {
tp[i+1] = 1-tp[s1];
tp[i] = tp[s1];
} else {
tp[i] = tp[i+1] = 1-tp[s1];
}
}
}
return accumulate(tp, tp+n, 0);
}
