#include "mushrooms.h"
int count_mushrooms(int n) {
if(n == 2){
int res = use_machine({0,1});
if(res) return 1;
else return 2;
}
int ans = 1;
int b = -1;
for(int i = 1; i<n; i+=2){
int res = use_machine({0,i,i+1});
if(res == 2) ans++;
else if(res == 0) ans+=2;
else{
if(use_machine({0,i})){
b = i+1;
}
else{
ans++; b = i+1;
}
}
}
int a = -1;
for(int i = b+1; i<n and b>=0; i+=2){
int res = use_machine({b,i,i+1});
if(res == 2) ans++;
else if(res == 1){
if(use_machine({b,i})){
ans+=2; a = i+1;
}
else{
ans++; a = i+1;
}
}
}
for(int i = a+1; i<n and a>=0; i+=2){
int res = use_machine({0,i,a,i+1});
if(res == 1 || res == 2){
ans++;
}
else if(res == 0) ans+=2;
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
0 ms |
344 KB |
Invalid value 4 in the query array. |
3 |
Halted |
0 ms |
0 KB |
- |