#define nDEBUG
#ifdef DEBUG
#define Printf(...) fprintf(stderr, __VA_ARGS__)
#else
#define Printf(...)
#endif
#include "elephants.h"
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long LL;
typedef vector<int> vi;
const int BLCK = 400; // sqrt n
int UPDCNTR = 0;
const int mxsz = 15e4 + 14;
int n;
LL l;
int pos[mxsz]; //ord[mxsz];
int wbl[mxsz]; // stores wbl block elephant i is in
int NUMBLOCKS;
int blocks[BLCK+10][2 * BLCK + 15];
int need[mxsz], nxt[mxsz];
int SZ[BLCK+10];
const int DMY = 15e4 + 5;
inline void workBlock(int B){
int sz = SZ[B];
if (!sz) return;
pos[DMY] = pos[blocks[B][sz-1]] + l + 1;
blocks[B][SZ[B]++] = DMY;
nxt[DMY] = DMY;
for (int i = sz-1, j; i >= 0; --i){
int u = blocks[B][i];
int le = i, ri = sz, md; j = sz;
while (le <= ri){
md = (le + ri) >> 1;
if (pos[blocks[B][md]] - pos[u] > l){j = md; ri = md-1;}
else le = md+1;
}
int v = blocks[B][j];
need[u] = need[v] + 1;
nxt[u] = v;
}
for (int i = sz-1; i >= 0; --i)
if (nxt[blocks[B][i]] == DMY) nxt[blocks[B][i]] = blocks[B][i];
else nxt[blocks[B][i]] = nxt[nxt[blocks[B][i]]];
SZ[B]--;
}
int tmp[mxsz];
inline void rebuild(){
int now = 0;
for (int B = 0; B < NUMBLOCKS; B++){
for (int i = 0; i < SZ[B]; i++)
tmp[now++] = blocks[B][i];
SZ[B] = 0;
}
for (int i = 0, B; i < n; i++){
// Printf( "%d%c", tmp[i], " \n"[i+1==n]);
B = i/BLCK;
blocks[B][SZ[B]++] = tmp[i];
wbl[tmp[i]] = B;
//ord[tmp[i]] = i;
}
for (int B = 0; B < NUMBLOCKS; B++){
workBlock(B);
}
#ifdef DEBUG
for (int i = 0; i < n; i++){
Printf( "%d: o_%d p_%d w_%d nd:%d nx:%d\n",
i, ord[i],pos[i],wbl[i],need[i], nxt[i]);
}
Printf("end rebuild\n");
#endif
}
inline int getAns(){
int x = -1; // next needed to cover
int ret = 0;
for (int i = 0; i < NUMBLOCKS; i++){
if (!SZ[i]) continue;
if (pos[blocks[i][SZ[i]-1]] <= x) continue;
int le = 0, ri = SZ[i] - 1, md, j = 0;
while (le <= ri){
md = (le + ri) >> 1;
// Printf( "%d %d %d\n", le, ri, md);
if (pos[blocks[i][md]] > x){j = md; ri = md-1;}
else le = md+1;
}
// Printf( "cur ret: %d x: %d b:%d nd:%d nxt:%d pos:%d\n",
// ret, x, blocks[i][j], need[blocks[i][j]],
// nxt[blocks[i][j]], pos[nxt[blocks[i][j]]]);
// Printf( "L :: %d\n", l);
ret += need[blocks[i][j]];
x = pos[nxt[blocks[i][j]]] + l;
// Printf( "new ret: %d x: %d\n", ret, x);
}
// Printf("END OF QUERY\n\n\n\n");
return ret;
}
void init(int N, int L, int X[]){
n = N; l = L;
for (int i = 0; i < n; i++) pos[i] = X[i];
NUMBLOCKS = ((n-1)/BLCK) + 1;
for (int i = 0, B; i < n; i++){
B = i/BLCK;
blocks[B][SZ[B]++] = i;
wbl[i] = B;
//ord[i] = i;
}
for (int B = 0; B < NUMBLOCKS; B++){
workBlock(B);
}
}
int update(int i, int y){
int w = wbl[i];
//blocks[w].erase(find(blocks[w].begin(), blocks[w].end(), i));
for (int p = 0; p < SZ[w]; p++){
if (blocks[w][p] == i){
for (int j = p; j < SZ[w] - 1; j++){
blocks[w][j] = blocks[w][j+1];
}
SZ[w]--; break;
}
}
workBlock(w);
pos[i] = y;
w = 0;
for (int B = 0; B < NUMBLOCKS; B++){
if (!SZ[B]) w++;
else if (y > pos[blocks[B][SZ[B]-1]]) w++;
else break;
}
if (w >= NUMBLOCKS) w--;
// Printf( "new block %d\n", w);
int idx = SZ[w];
blocks[w][idx] = i; SZ[w]++;
while (idx > 0 && pos[blocks[w][idx]] < pos[blocks[w][idx-1]]){
swap(blocks[w][idx], blocks[w][idx - 1]); idx--;
}
workBlock(w);
wbl[i] = w;
#ifdef DEBUG
Printf("post update\n");
for (int i = 0; i < n; i++){
Printf( "%d: o_%d p_%d w_%d nd:%d nx:%d\n",
i, ord[i],pos[i],wbl[i],need[i], nxt[i]);
}
fputs("\n", stderr);
for (int B = 0; B < NUMBLOCKS; B++){
for (int i : blocks[B]) Printf( "%d ", i);
}
fputs("\n", stderr);
Printf("end update\n\n");
#endif
if (++UPDCNTR >= BLCK+5){
UPDCNTR = 0;
rebuild();
#ifdef DEBUG
Printf( " REBUILD\n");
for (int i = 0; i < n; i++){
Printf( "%d: o_%d p_%d w_%d nd:%d nx:%d\n",
i, ord[i],pos[i],wbl[i],need[i], nxt[i]);
}
fputs("\n", stderr);
for (int B = 0; B < NUMBLOCKS; B++){
for (int i : blocks[B]) Printf( "%d ", i);
fputs("\n", stderr);
}
#endif
}
return getAns();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
715 ms |
1420 KB |
Output is correct |
8 |
Correct |
755 ms |
1656 KB |
Output is correct |
9 |
Correct |
811 ms |
2560 KB |
Output is correct |
10 |
Correct |
988 ms |
2680 KB |
Output is correct |
11 |
Correct |
890 ms |
2680 KB |
Output is correct |
12 |
Correct |
1422 ms |
2680 KB |
Output is correct |
13 |
Correct |
1144 ms |
2608 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
715 ms |
1420 KB |
Output is correct |
8 |
Correct |
755 ms |
1656 KB |
Output is correct |
9 |
Correct |
811 ms |
2560 KB |
Output is correct |
10 |
Correct |
988 ms |
2680 KB |
Output is correct |
11 |
Correct |
890 ms |
2680 KB |
Output is correct |
12 |
Correct |
1422 ms |
2680 KB |
Output is correct |
13 |
Correct |
1144 ms |
2608 KB |
Output is correct |
14 |
Correct |
1056 ms |
1884 KB |
Output is correct |
15 |
Correct |
1083 ms |
1920 KB |
Output is correct |
16 |
Correct |
2260 ms |
2808 KB |
Output is correct |
17 |
Correct |
2422 ms |
3548 KB |
Output is correct |
18 |
Correct |
2611 ms |
3576 KB |
Output is correct |
19 |
Correct |
1458 ms |
3448 KB |
Output is correct |
20 |
Correct |
2408 ms |
3448 KB |
Output is correct |
21 |
Correct |
2372 ms |
3448 KB |
Output is correct |
22 |
Correct |
1861 ms |
3468 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
715 ms |
1420 KB |
Output is correct |
8 |
Correct |
755 ms |
1656 KB |
Output is correct |
9 |
Correct |
811 ms |
2560 KB |
Output is correct |
10 |
Correct |
988 ms |
2680 KB |
Output is correct |
11 |
Correct |
890 ms |
2680 KB |
Output is correct |
12 |
Correct |
1422 ms |
2680 KB |
Output is correct |
13 |
Correct |
1144 ms |
2608 KB |
Output is correct |
14 |
Correct |
1056 ms |
1884 KB |
Output is correct |
15 |
Correct |
1083 ms |
1920 KB |
Output is correct |
16 |
Correct |
2260 ms |
2808 KB |
Output is correct |
17 |
Correct |
2422 ms |
3548 KB |
Output is correct |
18 |
Correct |
2611 ms |
3576 KB |
Output is correct |
19 |
Correct |
1458 ms |
3448 KB |
Output is correct |
20 |
Correct |
2408 ms |
3448 KB |
Output is correct |
21 |
Correct |
2372 ms |
3448 KB |
Output is correct |
22 |
Correct |
1861 ms |
3468 KB |
Output is correct |
23 |
Correct |
5887 ms |
6908 KB |
Output is correct |
24 |
Correct |
6222 ms |
6872 KB |
Output is correct |
25 |
Correct |
5195 ms |
6872 KB |
Output is correct |
26 |
Correct |
6504 ms |
6904 KB |
Output is correct |
27 |
Correct |
6015 ms |
6876 KB |
Output is correct |
28 |
Correct |
3866 ms |
2424 KB |
Output is correct |
29 |
Correct |
3939 ms |
2424 KB |
Output is correct |
30 |
Correct |
3874 ms |
2428 KB |
Output is correct |
31 |
Correct |
4169 ms |
2296 KB |
Output is correct |
32 |
Correct |
5039 ms |
6904 KB |
Output is correct |
33 |
Correct |
4494 ms |
6904 KB |
Output is correct |
34 |
Correct |
5979 ms |
6944 KB |
Output is correct |
35 |
Correct |
4062 ms |
6904 KB |
Output is correct |
36 |
Correct |
4684 ms |
7012 KB |
Output is correct |
37 |
Correct |
7433 ms |
6904 KB |
Output is correct |
38 |
Correct |
6321 ms |
6948 KB |
Output is correct |
39 |
Correct |
5174 ms |
7004 KB |
Output is correct |
40 |
Correct |
8052 ms |
6908 KB |
Output is correct |
41 |
Correct |
8819 ms |
6876 KB |
Output is correct |
42 |
Correct |
8447 ms |
6904 KB |
Output is correct |