Submission #1052305

#	Time	Username	Problem	Language	Result	Execution time	Memory
1052305		fuad27	Counting Mushrooms (IOI20_mushrooms)	C++17	80.14 / 100	5 ms	600 KiB

mushrooms

#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
int count_mushrooms(int n) {
	if(n == 2) {
		return 2-use_machine({0, 1});
	}
	int bb = min(283, n);
	int ss = use_machine({0, 1});
	int tt = use_machine({0, 2});
	vector<int> A, B;
	A.push_back(0);
	if(ss==0)
		A.push_back(1);
	else B.push_back(1);	
	if(tt==0)
		A.push_back(2);
	else B.push_back(2);	
	

	for(int i = 3;i<bb;i+=2) {
		if(i + 1 < bb) {
			if(A.size() >= 2) {
				int val = use_machine(vector{A[0], i, A[1], i+1});
				if(val < 2) {
					A.push_back(i);
				}
				else B.push_back(i);
				if(val%2 == 0) {
					A.push_back(i+1);
				}
				else B.push_back(i+1);
			}
			else {
				int val = use_machine(vector{B[0], i, B[1], i+1});
				if(val < 2) {
					B.push_back(i);
				}
				else A.push_back(i);
				if(val%2 == 0) {
					B.push_back(i+1);
				}
				else A.push_back(i+1);
			}
		}
		else {
			int val = use_machine(vector{A[0], i});
			if(val==0) {
				A.push_back(i);
			}
			else B.push_back(i);
		}
	}
	bool col=0;
	int szmx=A.size()-1;
	int ans=B.size();
	if(A.size() < B.size()){
		col=1;
		szmx = B.size()-1;
		ans=A.size();
	}
	for(int i = bb;i<n;i+=szmx) {
		vector<int> que;
		if(!col)
			que.push_back(A[0]);
		else 
			que.push_back(B[0]);
		for(int j = 0;j<min(n-i, szmx);j++) {
			que.push_back(i+j);
			if(!col)
				que.push_back(A[j+1]);
			else 
				que.push_back(B[j+1]);
		}
		ans+=use_machine(que)/2;
	}
	if(!col)
		return n-ans;
	return ans;
}

• Subtask 1: 80.1418 / 100