#include <bits/stdc++.h>
#include "advisor.h"
#define pb emplace_back
#define pii pair<int,int>
#define ff first
#define ss second
using namespace std;
#define debug(args...) kout("[ " + string(#args) + " ]", args)
void kout() { cerr << endl; }
template <class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ',kout(b...); }
template <class T> void pary(T L, T R) { while (L != R) cerr << *L << " \n"[++L==R]; }
const int MAXN = 100000;
int P;
vector <int> nxt[MAXN+1];
vector <int> cur;
vector <int> pos;
vector <int> id;
class Seg {
private:
pii arr[MAXN*4+5];
void pull(int now) {
arr[now] = max(arr[now*2], arr[now*2+1]);
}
public:
void build(int now, int l, int r) {
if (l == r) {
arr[now] = pii(*nxt[l].begin(), l);
return;
}
int mid = l + r >> 1;
build(now*2 , l, mid);
build(now*2+1,mid+1,r);
pull(now);
return;
}
void mdy(int p, int id, int now, int l, int r) {
if (l == p && r == p) {
arr[now] = pii(nxt[id][cur[id]], l);
return;
} else if (l > p || r < p) return;
int mid = l + r >> 1;
mdy(p, id, now*2 , l, mid);
mdy(p, id, now*2+1,mid+1,r);
pull(now);
return;
}
pii qry() {
return arr[1];
}
} seg;
void init(int *C, int N, int K, int M) {
P = __lg(N)+1;
id.resize(N);
cur.resize(N);
pos.resize(K);
for (int i = 0; i < N; i++) nxt[C[i]].pb(i);
for (int i = 0; i < N; i++) nxt[i].pb(N);
for (int i = 0; i < K; i++) pos[i] = i, id[i] = i;
for (int i = K; i < N; i++) id[i] = -1;
}
void Write(int num) {
for (int i = 0; i < P; i++) {
if ((1<<i) & num)
WriteAdvice(1);
else
WriteAdvice(0);
}
}
void ComputeAdvice(int *C, int N, int K, int M) {
init(C, N, K, M);
seg.build(1, 0, K-1);
for (int i = 0; i < N; i++) {
cur[C[i]]++;
if (id[C[i]] != -1) {
seg.mdy(id[C[i]], C[i], 1, 0, K-1);
} else {
auto [_t, p] = seg.qry();
Write(p);
id[pos[p]] = -1;
id[C[i]] = p;
pos[p] = C[i];
seg.mdy(p, C[i], 1, 0, K-1);
}
}
}
#include <bits/stdc++.h>
#include "assistant.h"
using namespace std;
#define debug(args...) kout("[ " + string(#args) + " ]", args)
void kout() { cerr << endl; }
template <class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ',kout(b...); }
template <class T> void pary(T L, T R) { while (L != R) cerr << *L << " \n"[++L==R]; }
const int MAXN = 100000;
int P;
vector <int> id;
vector <int> pos;
void init(int N, int K, int M) {
P = __lg(N)+1;
id.resize(N);
pos.resize(K);
for (int i = 0; i < K; i++) id[i] = i, pos[i] = i;
for (int i = K; i < N; i++) id[i] = -1;
}
int fnd(unsigned char *A, int a) {
int st = a * P, num = 0;
for (int i = st; i < st + P; i++) {
if ((int)A[i]) num += (1<<(i-st));
}
return num;
}
void Assist(unsigned char *A, int N, int K, int R) {
init(N, K, R);
int now = 0;
for (int i = 0; i < N; i++) {
int req = GetRequest();
if (id[req] == -1) {
int num = fnd(A, now++);
PutBack(pos[num]);
id[req] = id[pos[num]];
id[num] = -1;
pos[num] = req;
}
}
}
Compilation message
advisor.cpp: In member function 'void Seg::build(int, int, int)':
advisor.cpp:32:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
32 | int mid = l + r >> 1;
| ~~^~~
advisor.cpp: In member function 'void Seg::mdy(int, int, int, int, int)':
advisor.cpp:43:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
43 | int mid = l + r >> 1;
| ~~^~~
advisor.cpp: In function 'void ComputeAdvice(int*, int, int, int)':
advisor.cpp:82:12: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
82 | auto [_t, p] = seg.qry();
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
3288 KB |
Output is correct |
2 |
Correct |
3 ms |
3236 KB |
Output is correct |
3 |
Correct |
4 ms |
3268 KB |
Output is correct |
4 |
Incorrect |
9 ms |
3364 KB |
Error - Not putting back color when it is not on the scaffold |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
25 ms |
4360 KB |
Error - Not putting back color when it is not on the scaffold |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
226 ms |
12792 KB |
Output is correct |
2 |
Correct |
284 ms |
15156 KB |
Output is correct |
3 |
Correct |
282 ms |
15164 KB |
Output is correct |
4 |
Correct |
271 ms |
14704 KB |
Output is correct |
5 |
Incorrect |
216 ms |
13288 KB |
Error - Not putting back color when it is not on the scaffold |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
3536 KB |
Error - advice is too long |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
311 ms |
15656 KB |
Error - Not putting back color when it is not on the scaffold |
2 |
Correct |
293 ms |
15488 KB |
Output is partially correct - 841041 bits used |
3 |
Correct |
283 ms |
14996 KB |
Output is partially correct - 807466 bits used |
4 |
Correct |
291 ms |
15152 KB |
Output is partially correct - 806939 bits used |
5 |
Correct |
284 ms |
15248 KB |
Output is partially correct - 805358 bits used |
6 |
Correct |
288 ms |
15120 KB |
Output is partially correct - 807109 bits used |
7 |
Correct |
288 ms |
14992 KB |
Output is partially correct - 805902 bits used |
8 |
Correct |
284 ms |
15160 KB |
Output is partially correct - 808452 bits used |
9 |
Correct |
284 ms |
14996 KB |
Output is partially correct - 807874 bits used |
10 |
Incorrect |
382 ms |
17888 KB |
Error - Not putting back color when it is not on the scaffold |