Submission #600657

#	Time	Username	Problem	Language	Result	Execution time	Memory
600657		kig9981	Counting Mushrooms (IOI20_mushrooms)	C++17	91.50 / 100	9 ms	524 KiB

mushrooms

#include "mushrooms.h"
#include <bits/stdc++.h>
 
#ifdef NON_SUBMIT
#define TEST(n) (n)
#define tout cerr
#else
#define TEST(n) ((void)0)
#define tout cin
#endif
 
using namespace std;
 
vector<pair<int,int>> V;
vector<int> A, B, Q;
 
void query2()
{
	int a=V[0].first, b=V[1].first, r;
	if(A.size()>1) {
		r=use_machine({A[0],a,A[1],b});
		if(r&1) {
			B.push_back(b);
			A.push_back(V[1].second);
		}
		else {
			A.push_back(b);
			B.push_back(V[1].second);
		}
		if(r>1) {
			B.push_back(a);
			A.push_back(V[0].second);
		}
		else {
			A.push_back(a);
			B.push_back(V[0].second);
		}
	}
	else {
		r=use_machine({B[0],a,B[1],b});
		if(r&1) {
			A.push_back(b);
			B.push_back(V[1].second);
		}
		else {
			B.push_back(b);
			A.push_back(V[1].second);
		}
		if(r>1) {
			A.push_back(a);
			B.push_back(V[0].second);
		}
		else {
			B.push_back(a);
			A.push_back(V[0].second);
		}
	}
	V.clear();
}
 
void query3()
{
	int a, b, c, r;
	if(Q.size()+V.size()>=3) {
		a=Q.back(); Q.pop_back();
		b=Q.back(); Q.pop_back();
		c=Q.back(); Q.pop_back();
		if(A.size()>2) {
			r=use_machine({A[0],a,A[1],b,A[2],c});
			if(r&1) B.push_back(c);
			else A.push_back(c);
			r>>=1;
			if(r==0) {
				A.push_back(a);
				A.push_back(b);
			}
			else if(r==1) {
				V.emplace_back(a,b);
				if(V.size()>=2) query2();
			}
			else {
				B.push_back(a);
				B.push_back(b);
			}
		}
		else {
			r=use_machine({B[0],a,B[1],b,B[2],c});
			if(r&1) A.push_back(c);
			else B.push_back(c);
			r>>=1;
			if(r==0) {
				B.push_back(a);
				B.push_back(b);
			}
			else if(r==1) {
				V.emplace_back(a,b);
				if(V.size()>=2) query2();
			}
			else {
				A.push_back(a);
				A.push_back(b);
			}
		}
	}
	else if(Q.size()==1) {
		a=Q.back(); Q.pop_back();
		if(use_machine({0,a})) B.push_back(a);
		else A.push_back(a);
	}
	else if(Q.size()==2) {
		a=Q.back(); Q.pop_back();
		b=Q.back(); Q.pop_back();
		if(A.size()>1) {
			r=use_machine({A[0],a,A[1],b});
			if(r&1) B.push_back(b);
			else A.push_back(b);
			if(r>1) B.push_back(a);
			else A.push_back(a);
		}
		else {
			r=use_machine({B[0],a,B[1],b});
			if(r&1) A.push_back(b);
			else B.push_back(b);
			if(r>1) A.push_back(a);
			else B.push_back(a);
		}
	}
}
 
int query()
{
	int m=min(Q.size(),max(A.size(),B.size())), r;
	vector<int> I;
	if(A.size()>=m) {
		for(int i=0;i<m;i++) {
			I.push_back(A[i]); I.push_back(Q.back());
			Q.pop_back();
		}
		r=use_machine(I);
		if(r&1) B.push_back(I.back());
		else A.push_back(I.back());
		return m-1-(r>>1);
	}
	else {
		for(int i=0;i<m;i++) {
			I.push_back(B[i]); I.push_back(Q.back());
			Q.pop_back();
		}
		r=use_machine(I);
		if(r&1) A.push_back(I.back());
		else B.push_back(I.back());
		return r>>1;
	}
}
 
int count_mushrooms(int n)
{
	int ret=0;
	mt19937 mt(chrono::high_resolution_clock::now().time_since_epoch().count());
	if(n-1<=226) {
		ret=1;
		for(int i=1;i<n;i++) ret+=!use_machine({0,i});
		return ret;
	}
	for(int i=1;i<n;i++) Q.push_back(i);
	shuffle(Q.begin(),Q.end(),mt);
	A.push_back(0);
	while(max(A.size(),B.size())<3) {
		int c=Q.back();
		Q.pop_back();
		if(use_machine({0,c})) B.push_back(c);
		else A.push_back(c);
	}
	for(int i=0;i<40;i++) {
		if(Q.empty()) break;
		query3();
	}
	if(V.size()) {
		if(use_machine({0,V[0].first})) {
			B.push_back(V[0].first);
			A.push_back(V[0].second);
		}
		else {
			A.push_back(V[0].first);
			B.push_back(V[0].second);
		}
	}
	while(Q.size()) ret+=query();
	return ret+A.size();
}

Compilation message (stderr)

mushrooms.cpp: In function 'int query()':
mushrooms.cpp:134:13: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  134 |  if(A.size()>=m) {
      |     ~~~~~~~~^~~

• Subtask 1: 91.498 / 100