P4211 [LNOI2014]LCA LCT

P4211 [LNOI2014]LCA

链接

loj
luogu

思路

多次询问(sumlimits_{l leq i leq r}dep[LCA(i,z)])
可以转化成l到r上的点到根的路径+1
最后求一下1到z的路径和就是所求
区间([l,r])是可以差分的
离线直接求就行了。
树剖常数小，但还是比LCT多个log
我的LCT好慢啊

代码

#include <bits/stdc++.h>
#define ls c[x][0]
#define rs c[x][1]
using namespace std;
const int N = 1e5 + 7, mod = 201314;
int read() {
    int x = 0, f = 1; char s = getchar();
    for (;s > '9' || s < '0'; s = getchar()) if (s == '-') f = -1;
    for (;s >= '0' && s <= '9'; s = getchar()) x = x * 10 + s - '0';
    return x * f;        
}
int f[N], c[N][2], w[N], siz[N], sum[N], stak[N], lazy[N], lazytwo[N];
bool isroot(int x) {return c[f[x]][0] == x || c[f[x]][1] == x;}
void tag(int x){swap(ls,rs), lazy[x] ^= 1;}
void tagtwo(int x, int val) {
    sum[x] = (sum[x] + 1LL * val * siz[x] % mod) % mod;
    w[x] = (w[x] + val) % mod;
    lazytwo[x] = (lazytwo[x] + val) % mod;
}
void pushdown(int x) {
    if (lazy[x]) {
        if (ls) tag(ls);
        if (rs) tag(rs);
        lazy[x] ^= 1;
    }
    if (lazytwo[x]) {
        if (ls) tagtwo(ls, lazytwo[x]);
        if (rs) tagtwo(rs, lazytwo[x]);
        lazytwo[x] = 0; 
    }
}
void pushup(int x) {
    sum[x] = (sum[ls] + sum[rs] + w[x]) % mod;
    siz[x] = siz[ls] + siz[rs] + 1;
}
void rotate(int x) {
    int y = f[x], z = f[y], k = c[y][1] == x, w = c[x][!k];
    if (isroot(y)) c[z][c[z][1] == y] = x;
    c[x][!k] = y;
    c[y][k] = w;
    if (w) f[w] = y;
    f[x] = z;
    f[y] = x;
    pushup(y);
}
void splay(int x) {
    int y = x, z = 0;
    stak[++z] = y;
    while (isroot(y)) stak[++z] = y = f[y];
    while (z) pushdown(stak[z--]);
    while (isroot(x)) {
        y = f[x], z = f[y];
        if (isroot(y)) rotate((c[y][0] == x)^(c[z][0] == y) ? x : y);
        rotate(x);
    }
    pushup(x);
}
void access(int x) {
    for (int y = 0; x;x = f[y = x])
        splay(x), rs = y, pushup(x);
}
void makeroot(int x) {
    access(x), splay(x);
    tag(x);
}
int findroot(int x) {
    access(x), splay(x);
    while(ls) pushdown(x), x = ls;
    return x;
}
void split(int x, int y) {
    makeroot(x), access(y), splay(y);
}
void link(int x, int y) {
    makeroot(x);
    if (findroot(y) != x) f[x] = y;
}
void cut(int x, int y) {
    makeroot(x);
    if (findroot(y) == x && f[x] == y && !rs) {
        f[x] = c[y][0] = 0;
        pushup(y);
    }
}
struct node {
    int id, l, z, ans;
    node(int a = 0, int b = 0, int c = 0) {
        id = a, l = b, z = c;
    }
} Q[N];
bool cmp1(const node &a, const node& b) {
    return a.l < b.l;
}
bool cmp2(const node &a, const node& b) {
    return a.id < b.id;
}
int main() {
    // freopen("a.in", "r", stdin);
    int n = read(), m = read();
    for (int i = 2; i <= n; ++i) {
        int x = read() + 1;
        link(i, x);
    }
    for (int i = 1; i <= m; ++i) {
        int l = read() + 1, r = read() + 1, z = read() + 1;
        Q[i * 2 - 1] = node(i * 2 - 1, l - 1, z);
        Q[i * 2] = node(i * 2, r, z);
    }
    m <<= 1;
    sort(Q + 1, Q + 1 + m, cmp1);
    int js = 1;
    for (int i = 0; i <= n; ++i) {
        if (i) split(1, i),tagtwo(i, 1);
        while(Q[js].l == i) {
            split(1, Q[js].z);
            Q[js].ans = sum[Q[js].z];
            js++;
        }
    }
    sort(Q + 1, Q + 1 + m, cmp2);
    for (int i = 2; i <= m; i += 2) {
        int ans = Q[i].ans - Q[i - 1].ans;
        ans = (ans % mod + mod) % mod;
        printf("%d
", ans);
    }
    return 0;
}