Description

給出一個n個節點的有根樹（編號為0到n-1，根節點為0）。一個點的深度定義為這個節點到根的距離+1。
設dep[i]表示點i的深度，LCA(i,j)表示i與j的最近公共祖先。
有q次詢問，每次詢問給出l r z，求sigma_{l<=i<=r}dep[LCA(i,z)]。
（即，求在[l,r]區間內的每個節點i與z的最近公共祖先的深度之和）

Input

第一行2個整數n q。
接下來n-1行，分別表示點1到點n-1的父節點編號。
接下來q行，每行3個整數l r z。

Output

輸出q行，每行表示一個詢問的答案。每個答案對201314取模輸出

Sample Input

5 2
0
0
1
1
1 4 3
1 4 2

Sample Output

8
5

HINT

共5組數據，n與q的規模分別為10000,20000,30000,40000,50000。

Solution

答案要求：\(\sum_{i=l}^rdep[lca(i,z)]\)
轉化答案，我們把 \(l\) 到 \(r\) 中所有的點到根的路徑上的所有點權+1，然後答案就變成了 \(z\) 到根的權值之和
這個東西用LCT維護就行了
然後考慮多次詢問的問題
我們把詢問按 \(l\) （也不盡是 \(l\) ）從小到大排好序後，從1到n枚舉樹上的點加點權的時候，如果有一個詢問的 \(l\) 或 \(r\) 正好是當前枚舉的點，就記錄下來，也就是說，對於每個 \(l\)

#include<bits/stdc++.h>
#define ll long long
#define db double
#define ld long double
const int MAXN=50000+10,Mod=201314;
int n,q,ans[MAXN];
struct question{
    int id,pt,ps,vp;
    inline bool operator
 
 < (const question &A) const {
        return ps<A.ps;
    };
};
question Q[MAXN<<1];
#define lc(x) ch[(x)][0]
#define rc(x) ch[(x)][1]
struct LCT{
    int ch[MAXN][2],fa[MAXN],rev[MAXN],sum[MAXN],val[MAXN],add[MAXN],stack[MAXN],cnt,size[MAXN];
    inline bool nroot(int x)
    {
        return lc(fa[x])==x||rc(fa[x])==x;
    }
    inline void reverse(int x)
    {
        std::swap(lc(x),rc(x));
        rev[x]^=1;
    }
    inline void plus(int x,int k)
    {
        (sum[x]+=k*size[x]%Mod)%=Mod;
        (val[x]+=k)%=Mod;(add[x]+=k)%=Mod;
    }
    inline void pushup(int x)
    {
        size[x]=size[lc(x)]+size[rc(x)]+1;
        sum[x]=(sum[lc(x)]+sum[rc(x)]+val[x])%Mod;
    }
    inline void pushdown(int x)
    {
        if(add[x])
        {
            if(lc(x))plus(lc(x),add[x]);
            if(rc(x))plus(rc(x),add[x]);
            add[x]=0;
        }
        if(rev[x])
        {
            if(lc(x))reverse(lc(x));
            if(rc(x))reverse(rc(x));
            rev[x]=0;
        }
    }
    inline void rotate(int x)
    {
        int f=fa[x],p=fa[f],c=(rc(f)==x);
        if(nroot(f))ch[p][rc(p)==f]=x;
        fa[ch[f][c]=ch[x][c^1]]=f;
        fa[ch[x][c^1]=f]=x;
        fa[x]=p;
        pushup(f);
        pushup(x);
    }
    inline void splay(int x)
    {
        cnt=0;
        stack[++cnt]=x;
        for(register int i=x;nroot(i);i=fa[i])stack[++cnt]=fa[i];
        while(cnt)pushdown(stack[cnt--]);
        for(register int y=fa[x];nroot(x);rotate(x),y=fa[x])
            if(nroot(y))rotate((lc(y)==x)==(lc(fa[y])==y)?y:x);
        pushup(x);
    }
    inline void access(int x)
    {
        for(register int y=0;x;x=fa[y=x])splay(x),rc(x)=y,pushup(x);
    }
    inline void makeroot(int x)
    {
        access(x);splay(x);reverse(x);
    }
    inline void split(int x,int y)
    {
        makeroot(x);access(y);splay(y);
    }
    inline void link(int x,int y)
    {
        makeroot(x);fa[x]=y;
    }
};
LCT T;
#undef lc
#undef rc
template<typename T> inline void read(T &x)
{
    T data=0,w=1;
    char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char c='\0')
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
    if(c!='\0')putchar(c);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
int main()
{
    read(n);read(q);
    for(register int i=2;i<=n;++i)
    {
        int u;read(u);u++;
        T.link(i,u);
    }
    for(register int i=1,l,r,z;i<=q;++i)
    {
        read(l);read(r);read(z);
        l++;r++;z++;
        Q[i].ps=l-1;Q[i].pt=-1;Q[i].id=i;Q[i].vp=z;
        Q[i+q].ps=r;Q[i+q].pt=1;Q[i+q].id=i;Q[i+q].vp=z;
    }
    std::sort(Q+1,Q+q*2+1);
    for(register int i=1,j=1;i<=n;++i)
    {
        while(Q[j].ps<i&&j<=q*2)++j;
        T.split(1,i);T.plus(i,1);
        while(Q[j].ps==i&&j<=q*2)
        {
            T.split(1,Q[j].vp);
            (ans[Q[j].id]+=(Q[j].pt*T.sum[Q[j].vp]%Mod+Mod)%Mod)%=Mod;
            ++j;
        }
    }
    for(register int i=1;i<=q;++i)write(ans[i],'\n');
    return 0;
}

【刷題】BZOJ 3626 [LNOI2014]LCA