#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
typedef pair<ll,ll> pii;
const ll MAXN = 2e5+5;
const ll INF = 1e9+7;
int count_mushrooms(int n){
std::vector<int> m; int ans=1,i,j;
if(n==2){
m={0,1};
int c1 = use_machine(m);
if(c1==0)return 2;
return 1;
}
if(n<=220){
m={0,1};
int c1 = use_machine(m);
if(c1==0){//0 e 1 sao A
ans=2;
for(i=2;i<n-1;i+=2){//testar i e i+2
m={0,i,1,i+1};
int c1 = use_machine(m);
if(c1==0)ans+=2;
if(c1==1)ans++;
if(c1==2)ans++;
}
if(n%2==1){
m={0,n-1};
int c1 = use_machine(m);
if(c1==0)ans++;
}
}else{
m={1,2};
int c1 = use_machine(m);
if(c1==0){//1 e 2 sao B
ans=1;
for(i=3;i<n-1;i+=2){//testar i e i+2
m={1,i,2,i+1};
int c1 = use_machine(m);
if(c1==1)ans++;
if(c1==2)ans++;
if(c1==3)ans+=2;
}
if(n%2==0){
m={0,n-1};
int c1 = use_machine(m);
if(c1==0)ans++;
}
}else{//0 e 2 sao A
ans=2;
for(i=3;i<n-1;i+=2){//testar i e i+2
m={0,i,2,i+1};
int c1 = use_machine(m);
if(c1==0)ans+=2;
if(c1==1)ans++;
if(c1==2)ans++;
}
if(n%2==0){
m={0,n-1};
int c1 = use_machine(m);
if(c1==0)ans++;
}
}
}
return ans;
}
vector<int>b,a;
m={0,1};
int c1 = use_machine(m);
if(c1==0){//0 e 1 sao A
ans=2;
int go=0;
for(i=2;i<n-1;i+=2){//testar i e i+2
m={0,i,1,i+1};
int c1 = use_machine(m);
if(c1==0){
a.pb(i);
a.pb(i+1);
ans+=2;
}
if(c1==1){
a.pb(i);
b.pb(i+1);
ans++;
}
if(c1==2){
b.pb(i);
a.pb(i+1);
ans++;
}
if(c1==3){
b.pb(i);
b.pb(i+1);
}
if((int)a.size()>140){
go=1; i+=2;
break;
}
if((int)b.size()>140){
go=2; i+=2;
break;
}
}
if(go==1){//usar a's para avançar quick
int k=a.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(a[j]); m.pb(i+j);
}m.pb(a[round]);
int c1 = use_machine(m);
ans+=(round-(c1/2));
i+=round;
}
}
if(go==2){//usar b's para avançar quick
int k=b.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(b[j]); m.pb(i+j);
}m.pb(b[round]);
int c1 = use_machine(m);
ans+=(c1/2);
i+=round;
}
}
}else{
m={1,2};
int c1 = use_machine(m);
if(c1==0){//1 e 2 sao B
ans=1;
int go=0;
for(i=3;i<n-1;i+=2){//testar i e i+2
m={1,i,2,i+1};
int c1 = use_machine(m);
if(c1==0){
b.pb(i);
b.pb(i+1);
}
if(c1==1){
b.pb(i);
a.pb(i+1);
ans++;
}
if(c1==2){
a.pb(i);
b.pb(i+1);
ans++;
}
if(c1==3){
a.pb(i);
a.pb(i+1);
ans+=2;
}
if((int)a.size()>140){
go=1; i+=2;
break;
}
if((int)b.size()>140){
go=2; i+=2;
break;
}
}
if(go==1){//usar a's para avançar quick
int k=a.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(a[j]); m.pb(i+j);
}m.pb(a[round]);
int c1 = use_machine(m);
ans+=(round-(c1/2));
i+=round;
}
}
if(go==2){//usar b's para avançar quick
int k=b.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(b[j]); m.pb(i+j);
}m.pb(b[round]);
int c1 = use_machine(m);
ans+=(c1/2);
i+=round;
}
}
}else{//0 e 2 sao A
ans=2;
int go=0;
for(i=3;i<n-1;i+=2){//testar i e i+2
m={0,i,2,i+1};
int c1 = use_machine(m);
if(c1==0){
a.pb(i);
a.pb(i+1);
ans+=2;
}
if(c1==1){
a.pb(i);
b.pb(i+1);
ans++;
}
if(c1==2){
b.pb(i);
a.pb(i+1);
ans++;
}
if(c1==3){
b.pb(i);
b.pb(i+1);
}
if((int)a.size()>140){
go=1; i+=2;
break;
}
if((int)b.size()>140){
go=2; i+=2;
break;
}
}
//cout<<i<<
if(go==1){//usar a's para avançar quick
int k=a.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(a[j]); m.pb(i+j);
}m.pb(a[round]);
int c1 = use_machine(m);
ans+=(round-(c1/2));
i+=round;
}
}
if(go==2){//usar b's para avançar quick
int k=b.size()-1;
while(i<n){//testar i e os seus k-1 seguintes
int round = min(k,n-i);
m.clear();
for(j=0;j<round;j++){
m.pb(b[j]); m.pb(i+j);
}m.pb(b[round]);
int c1 = use_machine(m);
ans+=(c1/2);
i+=round;
}
}
}
}
return ans;
}
Compilation message
stations.cpp:1:10: fatal error: mushrooms.h: No such file or directory
1 | #include "mushrooms.h"
| ^~~~~~~~~~~~~
compilation terminated.
