Submission #432576

#	Time	Username	Problem	Language	Result	Execution time	Memory
432576		Pbezz	Counting Mushrooms (IOI20_mushrooms)	C++14	41.77 / 100	36 ms	360 KiB

mushrooms

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

using namespace std;

#define ll long long
#define pb push_back
typedef pair<ll,ll> pii;

const ll MAXN = 2e5+5;
const ll INF = 1e9+7;

int count_mushrooms(int n){
	std::vector<int> m; int ans=1,i,j;

	if(n==2){
	m={0,1};
	int c1 = use_machine(m);
	if(c1==0)return 2;
	return 1;
	}

	if(n<=200){
m={0,1};
	int c1 = use_machine(m);
	if(c1==0){//0 e 1 sao A
ans=2;
	for(i=2;i<n-1;i+=2){//testar i e i+2
	m={0,i,1,i+1};
	int c1 = use_machine(m);
	if(c1==0)ans+=2;
	if(c1==1)ans++;
	if(c1==2)ans++;
}
	if(n%2==1){
	m={0,n-1};
	int c1 = use_machine(m);
	if(c1==0)ans++;
}
	}else{
	m={1,2};
	int c1 = use_machine(m);
	if(c1==0){//1 e 2 sao B
ans=1;
	for(i=3;i<n-1;i+=2){//testar i e i+2
	m={1,i,2,i+1};
	int c1 = use_machine(m);
	if(c1==1)ans++;
	if(c1==2)ans++;
	if(c1==3)ans+=2;
}
	if(n%2==0){
	m={0,n-1};
	int c1 = use_machine(m);
	if(c1==0)ans++;
}
	}else{//0 e 2 sao A
ans=2;
	for(i=3;i<n-1;i+=2){//testar i e i+2
	m={0,i,2,i+1};
	int c1 = use_machine(m);
	if(c1==0)ans+=2;
	if(c1==1)ans++;
	if(c1==2)ans++;
}
	if(n%2==0){
	m={0,n-1};
	int c1 = use_machine(m);
	if(c1==0)ans++;
}
	}
}
	return ans;
}


vector<int>b,a;
	m={0,1};
	int c1 = use_machine(m);
	if(c1==0){//0 e 1 sao A
ans=2;
int go=0;
	for(i=2;i<n-1;i+=2){//testar i e i+2
	m={0,i,1,i+1};
	int c1 = use_machine(m);

	if(c1==0){
	a.pb(i);
	a.pb(i+1);
ans+=2;
	}
	if(c1==1){
	a.pb(i);
	b.pb(i+1);
ans++;
	}
	if(c1==2){
	b.pb(i);
	a.pb(i+1);
ans++;
	}
	if(c1==3){
	b.pb(i);
	b.pb(i+1);
	}

	if((int)a.size()>40){
	go=1; i+=2;
	break;
	}
	if((int)b.size()>40){
	go=2; i+=2;
	break;
	}

}

	if(go==1){//usar a's para avançar quick

	int k=a.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(a[j]); m.pb(i+j);
	}m.pb(a[round]);

	int c1 = use_machine(m);

	ans+=(round-(c1/2));
	i+=round;
	}
}

	if(go==2){//usar b's para avançar quick

	int k=b.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(b[j]); m.pb(i+j);
	}m.pb(b[round]);

	int c1 = use_machine(m);

	ans+=(c1/2);
	i+=round;
	}
}



	}else{
	m={1,2};
	int c1 = use_machine(m);
	if(c1==0){//1 e 2 sao B

ans=1;
int go=0;
	for(i=3;i<n-1;i+=2){//testar i e i+2
	m={1,i,2,i+1};
	int c1 = use_machine(m);

	if(c1==0){
	b.pb(i);
	b.pb(i+1);
	}
	if(c1==1){
	b.pb(i);
	a.pb(i+1);
ans++;
	}
	if(c1==2){
	a.pb(i);
	b.pb(i+1);
ans++;
	}
	if(c1==3){
	a.pb(i);
	a.pb(i+1);
ans+=2;
	}

	if((int)a.size()>40){
	go=1; i+=2;
	break;
	}
	if((int)b.size()>40){
	go=2; i+=2;
	break;
	}

}

	if(go==1){//usar a's para avançar quick

	int k=a.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(a[j]); m.pb(i+j);
	}m.pb(a[round]);

	int c1 = use_machine(m);

	ans+=(round-(c1/2));
	i+=round;
	}
}

	if(go==2){//usar b's para avançar quick

	int k=b.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(b[j]); m.pb(i+j);
	}m.pb(b[round]);

	int c1 = use_machine(m);

	ans+=(c1/2);
	i+=round;
	}
}




	}else{//0 e 2 sao A

ans=2;
int go=0;
	for(i=3;i<n-1;i+=2){//testar i e i+2
	m={0,i,2,i+1};
	int c1 = use_machine(m);

	if(c1==0){
	a.pb(i);
	a.pb(i+1);
ans+=2;
	}
	if(c1==1){
	a.pb(i);
	b.pb(i+1);
ans++;
	}
	if(c1==2){
	b.pb(i);
	a.pb(i+1);
ans++;
	}
	if(c1==3){
	b.pb(i);
	b.pb(i+1);
	}

	if((int)a.size()>40){
	go=1; i+=2;
	break;
	}
	if((int)b.size()>40){
	go=2; i+=2;
	break;
	}

}
//cout<<i<<
	if(go==1){//usar a's para avançar quick

	int k=a.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(a[j]); m.pb(i+j);
	}m.pb(a[round]);

	int c1 = use_machine(m);

	ans+=(round-(c1/2));
	i+=round;
	}
}

	if(go==2){//usar b's para avançar quick

	int k=b.size()-1;

	while(i<n){//testar i e os seus k-1 seguintes
	int round = min(k,n-i);

	m.clear();
	for(j=0;j<round;j++){
	m.pb(b[j]); m.pb(i+j);
	}m.pb(b[round]);

	int c1 = use_machine(m);

	ans+=(c1/2);
	i+=round;
	}
}





	}
}




	return ans;
}

• Subtask 1: 41.7745 / 100