제출 #1365762

#제출 시각아이디문제언어결과실행 시간메모리
1365762ByeWorldCounting Mushrooms (IOI20_mushrooms)C++20
0 / 100
2 ms432 KiB
#include "mushrooms.h"
#include <bits/stdc++.h>
#define ll long long
#define se second
#define fi first
#define pb push_back
#define lf (id<<1)
#define rg ((id<<1)|1)
#define md ((l+r)>>1)
#define USE use_machine
using namespace std;
typedef pair<int,int> pii;
typedef pair<pii,pii> ipii;
const int MAXN = 6e5+10;
const int MAXA = 5e4+10;
const int SQRT = 300;
const int INF = 2e9;
const int MOD = 1e9+87;
const int MOD2 = 1e9+7;
const int LOG = 30;

int n;
vector<int> a[2];
int ans = 0, nw = 1;

vector<int> m;
void ZER(){
	m.pb(a[0][0]); m.pb(nw++); m.pb(a[0][1]); m.pb(nw++);
	
	int ret = USE(m);
	if((ret>>1) & 1) a[1].pb(nw-2);
	else a[0].pb(nw-2);
	if(ret & 1) a[1].pb(nw-1);
	else a[0].pb(nw-1);
}
void ONE(){
	m.pb(a[1][0]); m.pb(nw++); m.pb(a[1][1]); m.pb(nw++);
	
	int ret = USE(m);
	if((ret>>1) & 1) a[0].pb(nw-2);
	else a[1].pb(nw-2);
	if(ret & 1) a[0].pb(nw-1);
	else a[1].pb(nw-1);
}

void pre(){
	while(nw <= 200){
		m.clear();
		// cout << nw << " nw\n";
		if(a[0].size() >= 3){
			m.pb(nw++); m.pb(a[0][0]); m.pb(nw++); m.pb(a[0][1]); m.pb(nw++); m.pb(a[0][2]);

			int ret = USE(m);
			if(ret & 1) a[1].pb(nw-3);
			else a[0].pb(nw-3);

			if(ret >= 4) a[1].pb(nw-1), a[1].pb(nw-2);
			else if(ret <= 1) a[0].pb(nw-1), a[0].pb(nw-2);
			else {
				m.clear();
				if(a[1].size() >= 2){
					m.pb(a[1][0]); m.pb(nw-2); m.pb(a[1][1]); 
					m.pb(a[0][0]); m.pb(nw-1); m.pb(a[0][1]); m.pb(nw++); m.pb(a[0][2]); m.pb(nw++);

					int ret = USE(m)-1;
					if(ret & 1) a[1].pb(nw-1);
					else a[0].pb(nw-1);
					if((ret>>1) & 1) a[1].pb(nw-2);
					else a[0].pb(nw-2);
					if((ret>>2) & 1) a[1].pb(nw-3), a[0].pb(nw-4);
					else a[0].pb(nw-3), a[1].pb(nw-4);

				} else {
					ZER();
				}
			}

		} else if(a[1].size() >= 3){
			m.pb(nw++); m.pb(a[1][0]); m.pb(nw++); m.pb(a[1][1]); m.pb(nw++); m.pb(a[1][2]);

			int ret = USE(m);
			if(ret & 1) a[0].pb(nw-3);
			else a[1].pb(nw-3);

			if(ret >= 4) a[0].pb(nw-1), a[0].pb(nw-2);
			else if(ret <= 1) a[1].pb(nw-1), a[1].pb(nw-2);
			else {
				m.clear();
				if(a[0].size() >= 2){
					m.pb(a[0][0]); m.pb(nw-2); m.pb(a[0][1]); 
					m.pb(a[1][0]); m.pb(nw-1); m.pb(a[1][1]); m.pb(nw++); m.pb(a[1][2]); m.pb(nw++);

					int ret = USE(m)-1;
					if(ret & 1) a[0].pb(nw-1);
					else a[1].pb(nw-1);
					if((ret>>1) & 1) a[0].pb(nw-2);
					else a[1].pb(nw-2);
					if((ret>>2) & 1) a[0].pb(nw-3), a[1].pb(nw-4);
					else a[1].pb(nw-3), a[0].pb(nw-4);

				} else {
					ZER();
				}
			}

		} else if(a[0].size() >= 2){
			ZER();

		} else if(a[1].size() >= 2){
			ONE();

		} else {
			m.pb(a[1][0]); m.pb(nw++);
			if(USE(m) == 1) a[0].pb(nw-1);
			else a[1].pb(nw-1);
		}
	}
}

int count_mushrooms(int N) {
	n = N;
	a[1] = {0};

	if(n >= 300) pre();
	ans = a[1].size();

	while(nw <= n-1){
		int tot = 0; 
		vector<int> m;

		if(a[0].size() < a[1].size()){//pake a
			for(int i=0; i<a[1].size(); i++){
				m.pb(a[1][i]); m.pb(nw++); 
				tot++;

				if(nw == n) break;
			}

			int ret = USE(m);
			ans += tot-1-ret/2; // ret/2 = B
			if(ret%2 == 1) a[0].pb(m.back()); // ada b di depan
			else ans++, a[1].pb(m.back());

		} else {//pake b
			for(int i=0; i<a[0].size(); i++){
				m.pb(a[0][i]); m.pb(nw++); 
				tot++;

				if(nw == n) break;
			}

			int ret = USE(m);
			ans += ret/2; // ret/2 = A
			if(ret%2 == 1) ans++, a[1].pb(m.back()); // ada b di depan
			else a[0].pb(m.back());
				
		}
	}

	return ans;
}
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…