#include "toxic.h"
#include <bits/stdc++.h>
using namespace std;
const int K = 8, T = 30;
vector<int> encode(vector<int> qry)
{
int n = qry.size();
vector<int> ret;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < (1 << i); j++)
{
ret.push_back(qry[i]);
}
}
return ret;
}
void determine_type(int n)
{
vector<int> ord(n);
iota(ord.begin(), ord.end(), 1);
random_shuffle(ord.begin(), ord.end());
string ans(n + 1, 'R');
int cnt = 0;
int tt = -1;
vector<int> stk;
int i = 0;
while (cnt < T && i < n)
{
int j = min(n, i + K);
int xx = -1;
{
vector<int> qry;
for (int k = i; k < j; k++)
{
qry.push_back(ord[k]);
}
int x = min(stk.size(), K - qry.size());
for (int k = 0; k < x; k++)
{
qry.push_back(stk.end()[-1 - k]);
}
xx = query_sample(encode(qry));
if (xx == (1 << qry.size()) - 1)
{
stk.insert(stk.end(), qry.begin(), qry.end());
i = j;
continue;
}
else
{
int y = qry.size() - (j - i);
for (int k = j - i; k < qry.size(); k++)
{
ans[qry[k]] = (xx >> k & 1) ? 'S' : 'R';
}
for (int k = 0; k < x; k++)
{
stk.pop_back();
}
}
}
auto make_query = [&](int mid)
{
vector<int> qry;
for (int k = i; k < mid; k++)
{
qry.push_back(ord[k]);
}
int x = min(stk.size(), K - qry.size());
for (int k = 0; k < x; k++)
{
qry.push_back(stk.end()[-1 - k]);
}
return qry;
};
int l = i, r = j;
while (l < r - 1)
{
int mid = (l + r) / 2;
vector<int> qry = make_query(mid);
int res = query_sample(encode(qry));
if (res == (1 << qry.size()) - 1)
{
l = mid;
}
else
{
int x = qry.size() - (mid - i);
for (int k = mid - i; k < qry.size(); k++)
{
ans[qry[k]] = (res >> k & 1) ? 'S' : 'R';
}
for (int k = 0; k < x; k++)
{
stk.pop_back();
}
r = mid;
}
}
vector<int> qry = make_query(l);
for (int k = 0; k < l - i; k++)
{
ans[qry[k]] = (xx >> k & 1) ? 'S' : 'R';
}
cerr << "ans " << ans << endl;
ans[ord[l]] = 'T';
cerr << "OK " << l << ' ' << ord[l] << endl;
i = l + 1;
tt = ord[l];
cnt++;
}
while (i < n)
{
stk.push_back(ord[i++]);
}
cerr << ans << endl;
for (int j = 0; j < stk.size(); j++)
{
cerr << stk[j] << endl;
}
cerr << "tt " << tt << endl;
while (stk.size())
{
int x = min((int)stk.size(), K);
vector<int> qry;
for (int k = 0; k < x; k++)
{
qry.push_back(stk.back());
stk.pop_back();
}
auto tmp = encode(qry);
tmp.push_back(tt);
int res = query_sample(tmp);
for (int k = 0; k < (int)qry.size(); k++)
{
ans[qry[k]] = (res >> k & 1) ? 'S' : 'R';
}
}
cerr << ans << endl;
for (int j = 1; j <= n; j++)
{
answer_type(j, ans[j]);
}
}
Compilation message
toxic.cpp: In function 'void determine_type(int)':
toxic.cpp:54:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
54 | for (int k = j - i; k < qry.size(); k++)
| ~~^~~~~~~~~~~~
toxic.cpp:53:9: warning: unused variable 'y' [-Wunused-variable]
53 | int y = qry.size() - (j - i);
| ^
toxic.cpp:91:29: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
91 | for (int k = mid - i; k < qry.size(); k++)
| ~~^~~~~~~~~~~~
toxic.cpp:119:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
119 | for (int j = 0; j < stk.size(); j++)
| ~~^~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
600 KB |
Output is correct |
2 |
Correct |
56 ms |
1004 KB |
Output is correct |
3 |
Partially correct |
57 ms |
1024 KB |
Partially correct |
4 |
Correct |
63 ms |
1020 KB |
Output is correct |
5 |
Correct |
49 ms |
1360 KB |
Output is correct |
6 |
Correct |
45 ms |
1360 KB |
Output is correct |
7 |
Correct |
45 ms |
1360 KB |
Output is correct |
8 |
Correct |
45 ms |
1348 KB |
Output is correct |
9 |
Partially correct |
48 ms |
1676 KB |
Partially correct |
10 |
Correct |
45 ms |
1364 KB |
Output is correct |
11 |
Partially correct |
46 ms |
1300 KB |
Partially correct |
12 |
Correct |
56 ms |
1108 KB |
Output is correct |
13 |
Correct |
56 ms |
1012 KB |
Output is correct |
14 |
Partially correct |
59 ms |
1180 KB |
Partially correct |
15 |
Correct |
47 ms |
1432 KB |
Output is correct |
16 |
Partially correct |
44 ms |
1364 KB |
Partially correct |
17 |
Correct |
49 ms |
1448 KB |
Output is correct |
18 |
Partially correct |
46 ms |
1404 KB |
Partially correct |
19 |
Partially correct |
45 ms |
1360 KB |
Partially correct |
20 |
Partially correct |
46 ms |
1408 KB |
Partially correct |
21 |
Partially correct |
45 ms |
1408 KB |
Partially correct |
22 |
Correct |
54 ms |
1128 KB |
Output is correct |
23 |
Correct |
55 ms |
852 KB |
Output is correct |
24 |
Partially correct |
57 ms |
1108 KB |
Partially correct |
25 |
Correct |
62 ms |
1360 KB |
Output is correct |
26 |
Correct |
55 ms |
1136 KB |
Output is correct |
27 |
Correct |
59 ms |
1080 KB |
Output is correct |
28 |
Correct |
55 ms |
892 KB |
Output is correct |
29 |
Correct |
49 ms |
1348 KB |
Output is correct |
30 |
Partially correct |
46 ms |
1464 KB |
Partially correct |
31 |
Correct |
45 ms |
1428 KB |
Output is correct |
32 |
Partially correct |
53 ms |
1420 KB |
Partially correct |
33 |
Partially correct |
48 ms |
1512 KB |
Partially correct |
34 |
Partially correct |
45 ms |
1520 KB |
Partially correct |
35 |
Partially correct |
44 ms |
1364 KB |
Partially correct |
36 |
Partially correct |
6 ms |
348 KB |
Partially correct |