#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;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);
}
else {
pbd.erase(pbd.find(ind));
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;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);
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);
}
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
504 KB |
Output is correct |
2 |
Correct |
0 ms |
508 KB |
Output is correct |
3 |
Correct |
2 ms |
776 KB |
Output is correct |
4 |
Incorrect |
3 ms |
792 KB |
Error - Putting back a color that is not on the scaffold |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
1372 KB |
Error - Putting back a color that is not on the scaffold |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
63 ms |
7248 KB |
Output is correct |
2 |
Correct |
80 ms |
9820 KB |
Output is correct |
3 |
Correct |
80 ms |
10024 KB |
Output is correct |
4 |
Correct |
81 ms |
9984 KB |
Output is correct |
5 |
Incorrect |
78 ms |
9480 KB |
Error - Putting back a color that is not on the scaffold |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
928 KB |
Output is correct |
2 |
Correct |
3 ms |
1052 KB |
Output is correct |
3 |
Incorrect |
1 ms |
648 KB |
Error - advice is too long |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
69 ms |
8492 KB |
Error - Putting back a color that is not on the scaffold |
2 |
Correct |
77 ms |
8600 KB |
Output is correct - 122000 bits used |
3 |
Correct |
83 ms |
8928 KB |
Output is correct - 125000 bits used |
4 |
Correct |
81 ms |
8708 KB |
Output is correct - 125000 bits used |
5 |
Correct |
82 ms |
8656 KB |
Output is correct - 125000 bits used |
6 |
Correct |
80 ms |
8784 KB |
Output is correct - 125000 bits used |
7 |
Correct |
79 ms |
8716 KB |
Output is correct - 124828 bits used |
8 |
Correct |
80 ms |
8656 KB |
Output is correct - 124910 bits used |
9 |
Correct |
82 ms |
8728 KB |
Output is correct - 125000 bits used |
10 |
Incorrect |
67 ms |
9516 KB |
Output isn't correct - not an optimal way |