Submission #346093

#	Time	Username	Problem	Language	Result	Execution time	Memory
346093		TheerapakG	Counting Mushrooms (IOI20_mushrooms)	C++17	92.62 / 100	9 ms	492 KiB

mushrooms

#include "mushrooms.h"
#include <bits/stdc++.h>

using namespace std;

vector<int> vec[2];

int count_mushrooms(int n) 
{
	if(n<3) return 2-use_machine({0, 1});
	else
	{
		vec[0].push_back(0);
		vec[use_machine({0, 1})].push_back(1);
		vec[use_machine({0, 2})].push_back(2);
		int s = vec[0].size();
		for(int i=3;i<n;)
		{
			int use = vec[0].size()>vec[1].size()?0:1;
			vector<int> query;
			int l = vec[use].size()>85?n:min(n,i+2);
			for(auto it=vec[use].begin(); i<l&&it!=vec[use].end(); i++, it++)
			{
				query.push_back(*it);
				query.push_back(i);
			}
			int ans = use_machine(query);
			if(query.size()==4) 
			{
				if(ans>1) vec[!use].push_back(query[1]);
				else vec[use].push_back(query[1]);
			}
			if(ans%2==0) vec[use].push_back(i-1); // last one is same
			else vec[!use].push_back(i-1); // last one is NOT same
			if(use==0) s+= (query.size()-ans)/2;
			else s+= (ans+1)/2;
		}
		return s;
	}
}

• Subtask 1: 92.623 / 100