QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#869396 | #9676. Ancestors | xcs321 | 0 | 2487ms | 253608kb | C++14 | 4.7kb | 2025-01-25 08:58:14 | 2025-01-25 08:58:15 |
Judging History
answer
#include<bits/stdc++.h>
#define int long long
#define N 2000005
using namespace std;
int n, m, B, root;
vector<int>p[N];
int dep[N], ans[N], dfn[N], len, tot;
struct RMQ_LCA { //O(n log n) - O(1) LCA
int z[N], r[N], lg[N], tot;
int f[N][17];
void dfs(int x, int fa) {
dep[x] = dep[fa] + 1;
r[++tot] = x;
dfn[x] = ++len;
z[x] = tot;
for(auto y : p[x]) {
dfs(y, x);
r[++tot] = x;
}
}
int Min(int x, int y) {
return dep[x] < dep[y] ? x : y;
}
void init() {
dfs(root, 0);
for(int i = 2; i <= tot; i++) lg[i] = lg[i / 2] + 1;
for(int i = 1; i <= tot; i++) f[i][0] = r[i];
for(int j = 1; j <= 15; j++)
for(int i = 1; i + (1 << j) - 1 <= tot; i++) f[i][j] = Min(f[i][j - 1], f[i + (1 << (j - 1)) - 1][j - 1]);
}
int LCA(int x, int y) {
x = z[x], y = z[y];
if(x > y) swap(x, y);
int h = lg[y - x + 1];
return Min(f[x][h], f[y - (1 << h) + 1][h]);
}
}ljm;
int tim1;
struct DS { //分块:O(1)修改, O(sqrt(n))查询
int v1[N], v2[N];
int X[N], Y[N];
int Get(int x) {
return ceil((double)x * 1.0 / B);
}
void update(int x, int y) {
X[++tim1] = x; Y[++tim1] = y;
v1[Get(x)] += y;
v2[x] += y;
}
int query(int L, int R) {
int res = 0;
for(int i = L; i <= R; ) {
if(i % B == 1 && i + B - 1 <= R) res += v1[Get(i)], i += B;
else res += v2[i], i++;
}
return res;
}
void Go(int x) { //撤销回到时间轴为x时
while(tim1 > x) {
v1[Get(X[tim1])] -= Y[tim1];
v1[X[tim1]] -= Y[tim1];
tim1--;
}
}
}ds;
struct List {
int Pre, Nxt;
}t[N];
struct node {
int l, r, x, id;
}z[N], h[N];
bool cmp1(node A, node B) {
return A.l < B.l;
}
bool cmp2(node A, node B) {
return A.r > B.r;
}
int v[N];
bool cmp(int x, int y) {
return dfn[x] < dfn[y];
}
vector<int>r[N];
void Prepare() {
for(int i = 1; i <= n; i++) v[i] = i;
sort(v + 1, v + n + 1, cmp);
for(int i = 1; i <= n; i++) r[dep[i]].push_back(v[i]);
for(int d = 1; d <= n; d++) {
for(int i = 0; i < r[d].size(); i++) {
if(i == 0) t[r[d][i]].Pre = -1;
else t[r[d][i]].Pre = r[d][i - 1];
if(i == r[d].size() - 1) t[r[d][i]].Nxt = -1;
else {
t[r[d][i]].Nxt = r[d][i + 1];
ds.update(d - dep[ljm.LCA(r[d][i], r[d][i + 1])], 1);
}
}
if(r[d].size()) ds.update(d, 1); //d深度只要有一个,贡献就要加1
}
}
int tim2;
int X[N], Y[N], opt[N];
void Go(int x) {
while(tim2 > x) {
if(opt[tim2] == 1) t[X[tim2]].Pre = Y[tim2];
else t[X[tim2]].Nxt = Y[tim2];
tim2--;
}
}
void Del(int x) {
if(t[x].Pre != -1) {
ds.update(dep[x] - dep[ljm.LCA(x, t[x].Pre)], -1);
tim2++; X[tim2] = t[x].Pre, Y[tim2] = t[t[x].Pre].Nxt, opt[tim2] = 2;
t[t[x].Pre].Nxt = t[x].Nxt;
}
if(t[x].Nxt != -1) {
ds.update(dep[x] - dep[ljm.LCA(x, t[x].Nxt)], -1);
tim2++; X[tim2] = t[x].Nxt, Y[tim2] = t[t[x].Nxt].Pre, opt[tim2] = 1;
t[t[x].Nxt].Pre = t[x].Pre;
}
if(t[x].Pre != -1 && t[x].Nxt != -1)
ds.update(dep[x] - dep[ljm.LCA(t[x].Pre, t[x].Nxt)], 1);
if(t[x].Pre == -1 && t[x].Nxt == -1) //这个深度删掉没有点了
ds.update(dep[x], -1);
}
void Sol(int L, int R) { //处理左端点在[L,R]的询问
sort(h + 1, h + tot + 1, cmp2);
for(int i = 1; i < L; i++) Del(i);
int curL = L, curR = n;
for(int i = 1; i <= tot; i++) {
while(curR > h[i].r) Del(curR--);
int lst1 = tim1, lst2 = tim2;
while(curL < h[i].l) Del(curL++);
ans[h[i].id] = ds.query(h[i].x + 1, n);
ds.Go(lst1); Go(lst2);
curL = L;
}
ds.Go(0); Go(0);
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> m; B = sqrt(n);
for(int i = 1, fa; i <= n; i++) {
cin >> fa;
if(fa) p[fa].push_back(i);
else root = i;
}
ljm.init(); Prepare();
for(int i = 1; i <= m; i++) cin >> z[i].l >> z[i].r >> z[i].x, z[i].id = i;
sort(z + 1, z + m + 1, cmp1);
int j = 0;
for(int i = 1; i <= n; i += B) {
tot = 0;
while(j + 1 <= m && z[j + 1].l <= i + B - 1) j++, h[++tot] = z[j];
Sol(i, min(n, i + B - 1));
}
for(int i = 1; i <= m; i++) cout << ans[i] << endl;
return 0;
}
/*
7 1
3 1 0 5 3 5 1
5 7 2
*/
//start : 2025/1/24 19:03
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 0
Wrong Answer
time: 12ms
memory: 124616kb
input:
7 5 3 1 0 5 3 5 1 1 3 1 5 7 2 1 5 1 4 7 1 4 7 2
output:
2 -1 2 1 0
result:
wrong answer 2nd numbers differ - expected: '1', found: '-1'
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Wrong Answer
Test #30:
score: 0
Wrong Answer
time: 622ms
memory: 154020kb
input:
50000 200000 42574 43129 47328 17982 40521 6668 12729 32377 201 11940 8599 11734 18349 41045 26854 22540 9897 33419 7463 1243 47272 27135 49050 49111 22435 42539 39924 20272 5843 9308 45963 3283 31185 13692 38952 20583 15885 24802 4773 953 49907 28689 36942 23550 19449 8970 33340 31665 5407 46023 18...
output:
11995 1321 -45772 -10918 -292924 -328348 -541298 -424565 -278852 -284861 -1122217 -984256 -663082 -49542 -606214 -208603 -1023285 -1141603 -393019 -690947 -1924623 -1478824 -1210920 -432031 -209851 -2099905 -1120261 -408583 -877196 -1647685 -870121 -1234718 -1987909 -920018 -326671 -181719 -3500448 ...
result:
wrong answer 1st numbers differ - expected: '12045', found: '11995'
Subtask #4:
score: 0
Skipped
Dependency #3:
0%
Subtask #5:
score: 0
Wrong Answer
Test #67:
score: 17
Accepted
time: 1992ms
memory: 253608kb
input:
100000 1000000 6457 23693 90928 23592 90440 75018 16865 3342 83718 16731 95103 31510 38719 27886 29093 41955 6596 46409 51839 10527 91993 61074 14405 34833 53674 42363 11490 43757 46191 6058 59164 96938 57858 40178 97523 84164 21582 72243 11267 47368 97058 6637 95208 60092 53943 16441 28363 64965 52...
output:
52956 18767 1319 13405 11021 455 50595 81481 6813 3640 58937 10991 70 4713 36 9517 39731 1166 67346 74637 2667 45182 4914 6774 1625 4198 52270 30435 60137 48654 29768 2815 6846 73091 21944 49693 9923 46795 29787 6866 2435 20773 2959 34666 4377 2428 4582 7527 38292 7253 3586 63817 28075 43828 20215 1...
result:
ok 1000000 numbers
Test #68:
score: 0
Wrong Answer
time: 2487ms
memory: 248228kb
input:
100000 1000000 82160 95864 48267 17482 19568 35077 14202 20440 4649 64145 148 2227 6969 39096 36508 20991 67700 90300 69215 57284 18492 9246 9629 7538 7845 30368 55600 48445 18542 41242 45052 25380 20894 91677 77040 73134 15572 21966 25343 14501 16227 23870 39207 50024 30787 11148 16884 63700 33205 ...
output:
57428 19901 2663 4024 53142 28120 24641 3164 43255 13020 4567 303 33353 67216 24140 34499 18991 31976 11954 17554 23889 7453 993 54540 3685 1422 59315 12938 4873 10064 9953 11026 48513 23709 22393 19299 54907 3459 12703 64901 1654 37424 9346 17841 6590 35296 14843 13507 3103 17787 40890 13466 8797 4...
result:
wrong answer 1st numbers differ - expected: '44469', found: '57428'
Subtask #6:
score: 0
Skipped
Dependency #1:
0%