#include "elephants.h"
#include <bits/stdc++.h>
const int MAXN = 1.5e5 + 11;
const int SQ = 411;
int n, l;
int x[MAXN];
using namespace std;
struct naive{
int n, y[SQ * 2], dp[SQ * 2], fin[SQ * 2];
void build(int n, int x[]){
naive::n = n;
for(int i = 0; i < n; i++){
y[i] = x[i];
}
}
void rem(int x){
int i; for(i = 0; i < n; i++) if(y[i] == x) break;
for(int j = i; j < n - 1; j++) y[j] = y[j + 1];
n--; precalc();
}
void add(int x){
int i; for(i = 0; i < n; i++) if(y[i] > x) break;
for(int j = n - 1; j >= i; j--) y[j + 1] = y[j];
y[i] = x;
n++; precalc();
}
void precalc(){
dp[n] = 0; fin[n] = n; int r = n - 1;
for(int j = n - 1; j >= 0; j--){
while(y[r] - y[j] > l) r--;
dp[j] = dp[r + 1] + 1;
fin[j] = r + 1 != n ? fin[r + 1] : j;
}
}
pair<int, int> qry(int x){
int id = upper_bound(y, y + n, x) - y;
return {dp[id], y[fin[id]]};
}
} cal[MAXN / SQ + 10];
int B[MAXN / SQ + 10], size_B;
void rebuild(){
std::sort(x, x + n);
B[0] = 0;
for(int i = 1; i < (n + SQ - 1) / SQ; i++){
B[i] = x[SQ * i];
}
B[(n + SQ - 1) / SQ] = 1 << 30;
size_B = (n + SQ - 1) / SQ;
for(int i = 0; i < (n + SQ - 1) / SQ; i++){
cal[i].build(min(SQ, n - i * SQ), x + i * SQ);
}
}
void rem(int x){
int id = upper_bound(B, B + size_B, x) - B - 1;
cal[id].rem(x);
}
void add(int x){
int id = upper_bound(B, B + size_B, x) - B - 1;
cal[id].add(x);
}
int qry(){
int ans = 0, x = -1, b = 0;
for(int b = 0; b < size_B; b++){
if(x >= B[b + 1]) continue;
pair<int, int> p = cal[b].qry(x);
ans += p.first; if(p.first != 0) x = p.second + l;
}
return ans;
}
void init(int N, int L, int X[]){
n = N; l = L; for(int i = 0; i < n; i++) x[i] = X[i]; rebuild();
}
int update(int i, int y){
rem(x[i]);
x[i] = y;
add(x[i]);
return qry();
}
Compilation message
elephants.cpp: In function 'int qry()':
elephants.cpp:69:26: warning: unused variable 'b' [-Wunused-variable]
69 | int ans = 0, x = -1, b = 0;
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
0 ms |
340 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
0 ms |
340 KB |
Output is correct |
7 |
Incorrect |
12 ms |
1156 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
0 ms |
340 KB |
Output is correct |
7 |
Incorrect |
12 ms |
1156 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
0 ms |
340 KB |
Output is correct |
7 |
Incorrect |
12 ms |
1156 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |