제출 #1047041

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1047041		MarwenElarbi	Counting Mushrooms (IOI20_mushrooms)	C++17	0 / 100	0 ms	344 KiB

mushrooms

#include <bits/stdc++.h>
using namespace std;
#include "mushrooms.h"
#define pb push_back
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int count_mushrooms(int n) {
	if(n==2) return 1+(1-use_machine({1,0}));
	vector<int> aa;
	aa.pb(0);
	vector<int> bb;
	int lst=3;
	( use_machine({0,1}) ? bb.pb(1) : aa.pb(1));
	( use_machine({0,2}) ? bb.pb(1) : aa.pb(1));
	int a=aa.size();int b=bb.size();
	for (int i = 3; i+1 < min(n,202); i+=2)
	{
		if(aa.size()==100||bb.size()==100) break;
		lst=min(200,i+2);
		int cur;
		if(aa.size()>=2){
			cur=use_machine({i,aa[0],i+1,bb[1]});
			if(cur==0){
				a+=2;
				aa.pb(i);
				aa.pb(i+1);
			}else if(cur==2){
				a++;
				b++;
				aa.pb(i);
				bb.pb(i+1);
			}else if(cur==3){
				b+=2;
				bb.pb(i);
				bb.pb(i+1);
			}	
		}
		else{
			cur=use_machine({i,bb[0],i+1,bb[1]});
			if(cur==0){
				b+=2;
				bb.pb(i);
				bb.pb(i+1);
			}else if(cur==2){
				a++;
				b++;
				bb.pb(i);
				aa.pb(i+1);
			}else if(cur==3){
				a+=2;
				aa.pb(i);
				aa.pb(i+1);
			}
		}
	}
	int ans=a;
	for (int i = lst; i < n; i+=99)
	{
		vector<int> cur;
		int k=0;
		cur.pb((aa.size()>bb.size() ? aa[k++] : bb[k++]));
		for (int j = 0; j < min(n-i,99); ++j)
		{
			cur.pb(j+i);
			cur.pb((aa.size()>bb.size() ? aa[k++] : bb[k++]));
		}
		int cnt=use_machine(cur);
		ans+=(aa.size()<=bb.size() ? cnt/2 : cur.size()-k-cnt/2);
	}
	return ans;
}

• Subtask 1: 0 / 100