#include "toxic.h"
#include <bits/stdc++.h>
using namespace std;
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()
#define show(x) cerr << #x << " is " << x << endl;
#define show2(x,y) cerr << #x << " is " << x << " " << #y << " is " << y << endl;
typedef pair<int, int> ii;
vector<int> toquery;
int query(){
return query_sample(toquery);
}
deque<int> an;
deque<int> nt;
deque<int> unknown;
int onetoxic = -1;
set<int> antidode;
bool checkunknown(vector<int> elements){
if(sz(elements) == 0) return false;
toquery.clear();
for(int i = 0;i < sz(elements);i++){
for(int j = 0;j < (1 << i);j++) toquery.push_back(elements[i]);
}
int cnt = min(sz(an), 8-sz(elements));
for(int i = 0;i < cnt;i++){
for(int j = 0;j < (1 << (i+sz(elements)));j++) toquery.push_back(an[i]);
}
int res = query_sample(toquery);
if(res == sz(toquery)){
return false;
}
else{
for(int i = 0;i < sz(elements);i++){
if(res & (1 << i)){
answer_type(elements[i], 'S');
antidode.insert(elements[i]);
}
}
for(int i = 0;i < cnt;i++){ ///piggybacking
if(res & (1 << (i+sz(elements)))) answer_type(an[0], 'S');
else answer_type(an[0], 'R');
an.pop_front();
}
return true;
}
}
bool checknt(vector<int> elements){
if(sz(elements) == 0) return false;
toquery.clear();
for(int i = 0;i < sz(elements);i++){
toquery.push_back(elements[i]);
}
int cnt = min(sz(an), 8);
for(int i = 0;i < cnt;i++){
for(int j = 0;j < (1 << i);j++) toquery.push_back(an[i]);
}
int res = query_sample(toquery);
if(res == sz(toquery)){
return false;
}
else{
for(int i = 0;i < cnt;i++){ ///piggybacking
if(res & (1 << i)) answer_type(an[0], 'S');
else answer_type(an[0], 'R');
an.pop_front();
}
return true;
}
}
void bsta(vector<int> elements){ ///assuming inside this group already got something bad
if(sz(elements) == 0) return;
if(sz(elements) == 1){
onetoxic = elements[0];
answer_type(elements[0], 'T');
return;
}
vector<int> newelements;
for(int x : elements) if(antidode.find(x) == antidode.end()) newelements.push_back(x);
int mid = sz(newelements) / 2;
vector<int> L, R;
for(int i = 0;i < mid;i++){
L.push_back(newelements[i]);
}
for(int i = mid;i < sz(newelements);i++){
R.push_back(newelements[i]);
}
if(checknt(L)){ ///L have toxic
for(int x : R) nt.push_back(x);
bsta(L);
}
else{ ///L don't have toxic
for(int x : L) answer_type(x, 'R');
bsta(R);
}
}
void determine_type(int n){
an.clear(); unknown.clear(); antidode.clear();
for(int i = 1;i <= n;i++) unknown.push_back(i);
random_shuffle(all(unknown));
while(sz(unknown) > 0){
//show(sz(unknown));
int batchsize = min(7, sz(unknown));
vector<int> totest;
for(int i = 0;i < batchsize;i++){
totest.push_back(unknown[0]);
unknown.pop_front();
}
if(checkunknown(totest)) bsta(totest);
else{
for(int x : totest) an.push_back(x);
}
}
while(sz(nt) > 0){
int batchsize = min(8, sz(nt));
vector<int> totest;
for(int i = 0;i < batchsize;i++){
totest.push_back(nt[0]);
nt.pop_front();
}
if(checknt(totest)) bsta(totest);
else{
for(int x : totest) answer_type(x, 'R');
}
}
//if(current_tc % 100 == 0) show(sz(an));
while(sz(an) > 0){
checknt({onetoxic});
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
9 ms |
340 KB |
Output is correct |
3 |
Correct |
12 ms |
376 KB |
Output is correct |
4 |
Correct |
9 ms |
376 KB |
Output is correct |
5 |
Correct |
8 ms |
376 KB |
Output is correct |
6 |
Correct |
9 ms |
372 KB |
Output is correct |
7 |
Correct |
9 ms |
320 KB |
Output is correct |
8 |
Correct |
9 ms |
396 KB |
Output is correct |
9 |
Correct |
10 ms |
396 KB |
Output is correct |
10 |
Correct |
13 ms |
340 KB |
Output is correct |
11 |
Correct |
10 ms |
468 KB |
Output is correct |
12 |
Correct |
9 ms |
340 KB |
Output is correct |
13 |
Correct |
15 ms |
312 KB |
Output is correct |
14 |
Correct |
10 ms |
408 KB |
Output is correct |
15 |
Correct |
9 ms |
376 KB |
Output is correct |
16 |
Correct |
9 ms |
340 KB |
Output is correct |
17 |
Correct |
9 ms |
340 KB |
Output is correct |
18 |
Correct |
10 ms |
340 KB |
Output is correct |
19 |
Correct |
10 ms |
400 KB |
Output is correct |
20 |
Correct |
10 ms |
340 KB |
Output is correct |
21 |
Correct |
10 ms |
404 KB |
Output is correct |
22 |
Correct |
10 ms |
336 KB |
Output is correct |
23 |
Correct |
8 ms |
340 KB |
Output is correct |
24 |
Correct |
10 ms |
340 KB |
Output is correct |
25 |
Correct |
9 ms |
360 KB |
Output is correct |
26 |
Correct |
9 ms |
376 KB |
Output is correct |
27 |
Correct |
9 ms |
340 KB |
Output is correct |
28 |
Correct |
9 ms |
340 KB |
Output is correct |
29 |
Correct |
11 ms |
376 KB |
Output is correct |
30 |
Correct |
11 ms |
340 KB |
Output is correct |
31 |
Correct |
10 ms |
340 KB |
Output is correct |
32 |
Correct |
10 ms |
376 KB |
Output is correct |
33 |
Correct |
11 ms |
340 KB |
Output is correct |
34 |
Correct |
8 ms |
400 KB |
Output is correct |
35 |
Correct |
9 ms |
340 KB |
Output is correct |
36 |
Correct |
1 ms |
212 KB |
Output is correct |