QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#869396#9676. Ancestorsxcs3210 2487ms253608kbC++144.7kb2025-01-25 08:58:142025-01-25 08:58:15

Judging History

This is the latest submission verdict.

  • [2025-01-25 08:58:15]
  • Judged
  • Verdict: 0
  • Time: 2487ms
  • Memory: 253608kb
  • [2025-01-25 08:58:14]
  • Submitted

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%