Submission #800629

#	Time	Username	Problem	Language	Result	Execution time	Memory
800629		QwertyPi	Counting Mushrooms (IOI20_mushrooms)	C++14	100 / 100	7 ms	344 KiB

mushrooms

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

std::vector<int> a[2];
int b[2];
 
int id, n;
void query_1(){
	if(id >= n) return;
	std::vector<int> m {0, id};
	int c = use_machine(m);
	a[c].push_back(id);
	id++;
}
 
void query_2a(){
	if(id >= n) return;
	if(a[0].size() <= 1 && a[1].size() <= 1 || id == n - 1){ query_1(); return; }
	int majority = a[0].size() >= a[1].size() ? 0 : 1;
	std::vector<int> m = {a[majority][0], id, a[majority][1], id + 1};
	int c = use_machine(m);
	id += 2;
	a[majority ^ (c / 2)].push_back(m[1]);
	a[majority ^ (c % 2)].push_back(m[3]);
}
 
void query_2b(){ // 7 / 3 bits per query
	if(id >= n) return;
	if(a[0].size() <= 3 && a[1].size() <= 3 || n - id <= 7){ query_2a(); return; }
	int majority = a[0].size() >= a[1].size() ? 0 : 1;
	std::vector<int> x {id + 3, id + 4, id + 5, id + 6};
	std::vector<int> m1{id + 0, a[majority][0], x[0], a[majority][1], x[1], a[majority][2], x[2], a[majority][3]};
	std::vector<int> m2{id + 1, a[majority][0], x[0], a[majority][1], x[1], a[majority][2], x[3], a[majority][3]};
	std::vector<int> m3{id + 2, a[majority][0], x[0], a[majority][1], x[2], a[majority][2], x[3], a[majority][3]};
	int c1 = use_machine(m1), c2 = use_machine(m2), c3 = use_machine(m3);
	a[majority ^ (c1 % 2)].push_back(id + 0);
	a[majority ^ (c2 % 2)].push_back(id + 1);
	a[majority ^ (c3 % 2)].push_back(id + 2);
	for(int mask = 0; mask < 16; mask++){
		std::vector<int> c;
		for(int j = 0; j < 4; j++){
			c.push_back((mask & (1 << j)) > 0);
		}
		if(c[0] + c[1] + c[2] == c1 / 2 && c[0] + c[1] + c[3] == c2 / 2 && c[0] + c[2] + c[3] == c3 / 2){
			for(int j = 0; j < 4; j++){
				a[majority ^ c[j]].push_back(x[j]);
			}
		}
	}
	id += 7;
}
 
void query_2c(){ // 5 / 2 bits per query
	if(id >= n) return;
	int majority = a[0].size() >= a[1].size() ? 0 : 1;
	if(a[majority].size() < 3 || a[!majority].size() < 2 || n - id <= 5){ query_2a(); return; }
	std::vector<int> x {id + 2, id + 3, id + 4};
	std::vector<int> m1{id + 0, a[majority][0], x[0], a[majority][1], x[1], a[majority][2], a[!majority][0], x[2], a[!majority][1]};
	std::vector<int> m2{id + 1, a[majority][0], x[0], a[majority][1], x[2], a[majority][2]};
	int c1 = use_machine(m1), c2 = use_machine(m2); --c1;
	a[majority ^ (c1 % 2)].push_back(id + 0);
	a[majority ^ (c2 % 2)].push_back(id + 1);
	c1 /= 2, c2 /= 2;
	std::vector<int> c;
	if(c1 == 3) c = {1, 1, 0};
	else if(c1 == 2) {
		if(c2 == 2) c = {1, 1, 1};
		else if(c2 == 1) c = {1, 0, 0};
		else c = {0, 1, 0};
	}else if(c1 == 1){
		if(c2 == 2) c = {1, 0, 1};
		else if(c2 == 1) c = {0, 1, 1};
		else c = {0, 0, 0};
	}else{
		c = {0, 0, 1};
	}
	for(int i = 0; i < 3; i++){
		a[majority ^ c[i]].push_back(x[i]);
	}
	id += 5;
}
void query_3(){
	if(id >= n) return;
	if(a[0].size() <= 1 && a[1].size() <= 1){ query_1(); return; }
	int majority = a[0].size() >= a[1].size() ? 0 : 1;
	int k = a[majority].size(); k = std::min(k, n - id);
	std::vector<int> m;
	for(int i = id; i < id + k; i++){
		m.push_back(a[majority][i - id]); m.push_back(i);
	}
	int c = use_machine(m);
	id += k;
	a[majority ^ (c % 2)].push_back(m.back());
	b[majority ^ 1] += c / 2;
	b[majority] += (k - 1) - c / 2;
}
 
int count_mushrooms(int N) {
	a[0].clear(); a[1].clear(); b[0] = b[1] = 0;
	n = N; a[0].push_back(0); id = 1;
	if(n <= 1000){
		while(id < n) query_3();
		return a[0].size() + b[0];
	}
	const int SQ = 196;
	while(id < SQ) query_2c();
	while(id < n) query_3();
	return a[0].size() + b[0];
}

Compilation message (stderr)

mushrooms.cpp: In function 'void query_2a()':
mushrooms.cpp:18:22: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   18 |  if(a[0].size() <= 1 && a[1].size() <= 1 || id == n - 1){ query_1(); return; }
      |     ~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~
mushrooms.cpp: In function 'void query_2b()':
mushrooms.cpp:29:22: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   29 |  if(a[0].size() <= 3 && a[1].size() <= 3 || n - id <= 7){ query_2a(); return; }
      |     ~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~

• Subtask 1: 100 / 100