#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include "advisor.h"
using namespace std;
using namespace __gnu_pbds;
using Set = tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update>;
void ComputeAdvice(int *C, int N, int K, int M) {
set<pair<int, int>> q;
int s[N], nx[N];
bool f[N] = {false};
fill(s, s+N, N);
int c[N];
for(int i = N-1;i >= 0;i--) {
nx[i] = s[C[i]];
s[C[i]] = i;
}
int sz = 0;
Set pbd;
vector<int> bad;
for(int i = 0;i < K;i++){
WriteAdvice(s[i] != N);
if(s[i] != N) {
sz++;
pbd.insert(i);
q.insert({s[i], i});
} else {
bad.push_back(i);
}
f[i] = true;
c[i] = i;
}
for(int i = 0;i < N-1-bad.size();i++){
if(f[C[i]]) {
int ind = q.begin()->second;
q.erase(q.begin());
WriteAdvice(nx[i] != N);
if(nx[i] == N) {
sz--;
bad.push_back(ind);
pbd.erase(pbd.find(ind));
}
else {
q.insert({nx[i], ind});
}
continue;
}
if(!bad.empty()){
int ind = bad.back();
bad.pop_back();
WriteAdvice(nx[i] != N);
if(nx[i] != N) {
sz++;
pbd.insert(ind);
q.insert({nx[i], ind});
} else bad.push_back(ind);
f[c[ind]] = false;
f[C[i]] = true;
c[ind] = C[i];
continue;
}
int ind = q.rbegin()->second;
int L = log2(sz-1) + 1;
int pind = pbd.order_of_key(ind);
for(int b = 0;b < L;b++) WriteAdvice(1&(pind>>b));
WriteAdvice(nx[i] != N);
q.erase(prev(q.end()));
if(nx[i] == N) {
sz--;
pbd.erase(pbd.find(ind));
bad.emplace_back(ind);
}
else q.insert({nx[i], ind});
f[c[ind]] = false;
f[C[i]] = true;
c[ind] = C[i];
}
}
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include "assistant.h"
using namespace std;
using namespace __gnu_pbds;
using Set = tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update>;
void Assist(unsigned char *A, int N, int K, int R) {
int c[N], indo[N];
bool f[N] = {false};
int cur = 0, sz = 0;
Set pbd;
vector<pair<int, int>> bad;
for(int i = 0;i < K;i++){
f[i] = true;
c[i] = i;
indo[i] = i;
if(A[cur++]){
sz++;
pbd.insert(i);
} else bad.emplace_back(i, i);
}
for(int i = 0;i < N-1-bad.size();i++){
int r = GetRequest();
if(f[r]) {
if(!A[cur++]){
sz--;
pbd.erase(pbd.find(indo[r]));
bad.emplace_back(r, indo[r]);
}
continue;
}
if(!bad.empty()){
PutBack(bad.back().first);
int ind = bad.back().second;
f[r] = true;
f[bad.back().first] = false;
indo[r] = ind;
c[ind] = r;
bad.pop_back();
if(A[cur++]) {
sz++;
pbd.insert(ind);
} else bad.emplace_back(r, ind);
continue;
}
int pind = 0, L = log2(sz-1)+1;
for(int b = 0;b < L;b++) {
if(A[cur]) pind |= 1 << b;
cur++;
}
int ind = *pbd.find_by_order(pind);
PutBack(c[ind]);
f[r] = true;
f[c[ind]] = false;
c[ind] = r;
indo[r] = ind;
if(!A[cur++]){
sz--;
pbd.erase(pbd.find(ind));
bad.emplace_back(r, ind);
}
}
int r;
for(auto p : bad){
r = GetRequest();
if(f[r]) continue;
PutBack(p.first);
f[r] = true;
}
r = GetRequest();
if(f[r]) return;
for(int i = 0;i < N;i++) if(f[i]) {
PutBack(i);
return;
}
}
Compilation message
advisor.cpp: In function 'void ComputeAdvice(int*, int, int, int)':
advisor.cpp:33:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
33 | for(int i = 0;i < N-1-bad.size();i++){
| ~~^~~~~~~~~~~~~~~~
assistant.cpp: In function 'void Assist(unsigned char*, int, int, int)':
assistant.cpp:23:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
23 | for(int i = 0;i < N-1-bad.size();i++){
| ~~^~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
508 KB |
Output is correct |
2 |
Correct |
0 ms |
516 KB |
Output is correct |
3 |
Incorrect |
1 ms |
768 KB |
Error - Too many requests |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
1356 KB |
Output is correct |
2 |
Correct |
51 ms |
4148 KB |
Output is correct |
3 |
Correct |
91 ms |
11260 KB |
Output is correct |
4 |
Correct |
79 ms |
6508 KB |
Output is correct |
5 |
Correct |
90 ms |
7048 KB |
Output is correct |
6 |
Incorrect |
95 ms |
7348 KB |
Error - Putting back a color that is not on the scaffold |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
63 ms |
7140 KB |
Output is correct |
2 |
Correct |
78 ms |
8696 KB |
Output is correct |
3 |
Correct |
83 ms |
8744 KB |
Output is correct |
4 |
Incorrect |
84 ms |
8748 KB |
Error - Too many requests |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
924 KB |
Output is correct |
2 |
Correct |
4 ms |
968 KB |
Output is correct |
3 |
Incorrect |
1 ms |
636 KB |
Error - advice is too long |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
82 ms |
8440 KB |
Error - Too many requests |
2 |
Correct |
79 ms |
8648 KB |
Output is correct - 108451 bits used |
3 |
Correct |
94 ms |
8740 KB |
Output is correct - 109675 bits used |
4 |
Incorrect |
86 ms |
8728 KB |
Error - Too many requests |
5 |
Incorrect |
80 ms |
8684 KB |
Error - Putting back a color that is not on the scaffold |
6 |
Correct |
87 ms |
8724 KB |
Output is correct - 109566 bits used |
7 |
Incorrect |
93 ms |
8644 KB |
Error - Too many requests |
8 |
Correct |
80 ms |
8772 KB |
Output is correct - 109657 bits used |
9 |
Correct |
91 ms |
8816 KB |
Output is correct - 109630 bits used |
10 |
Incorrect |
74 ms |
9436 KB |
Error - Putting back a color that is not on the scaffold |