#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
int count_mushrooms(int n) {
	vector<int> m, a, b;
	m.push_back(0);
	a.push_back(0);
	int fin = 1;
	for (int i = 1; i < n; i++) {
		m.push_back(i);
		int res = use_machine(m);
		if (res == 0) a.push_back(i);
		else b.push_back(i);
		m.pop_back();
		fin++;
		if (a.size() > n/500 || b.size() > n/500) break;
	}
	
	vector<int> majo = (a.size() > b.size() ? a : b);
	int ans;
	if (a.size() > b.size()) ans = b.size();
	else ans = a.size();
	while (fin != n) {
		int idx = 0;
		vector<int> rq;
		while (idx < majo.size() && fin != n) {
			rq.push_back(fin);
			rq.push_back(majo[idx]);
			fin++;
			idx++;
		}
		ans += (use_machine(rq)+1)/2;
	}
	if (a.size() > b.size()) return n-ans;
	else return ans;
}
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... |