[LNOI2014]LCA

[LNOI2014]LCA

题目

给出一个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

OUTPUT

8

5

解题报告

首先提出一个结论

$x$点与$y$点的$LCA$可以由如下方式求得：

将$x$点到根路径上所有点点权置为$1$，其余点点权置为$0$，则$y$点到根路径上点权和即为所求

原因很简单，手画一下就可以发现这个规律

有了这个结论，我们就可以把求$LCA$这种操作转化为路径上的点权操作了，也就是说，树链剖分就可行了起来

我们考虑如何处理一段区间的贡献，显然可以用差分来搞，那么我们就可以将操作离线，将询问拆成两个，进行差分，扫一遍节点，进行点权覆盖与差分查询即可

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
inline int read(){
    int sum(0);char ch(getchar());
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';sum=sum*10+(ch^48),ch=getchar());
    return sum;
}
const int mod(201314);
struct edge{int e;edge *n;}*pre[50005];
inline void insert(int s,int e){edge *tmp(new edge);tmp->e=e;tmp->n=pre[s];pre[s]=tmp;}
int n,q;
int z[50005];
vector<int>sub[50005],add[50005];
int sum[200005],lazy[200005];
inline void pushup(int i){sum[i]=(sum[i<<1]+sum[i<<1|1])%mod;}
inline void pushdown(int i,int len){
    if(lazy[i]){
        lazy[i<<1]=(lazy[i<<1]+lazy[i])%mod;
        lazy[i<<1|1]=(lazy[i<<1|1]+lazy[i])%mod;
        sum[i<<1]=(sum[i<<1]+lazy[i]*(len-(len>>1))%mod)%mod;
        sum[i<<1|1]=(sum[i<<1|1]+lazy[i]*(len>>1)%mod)%mod;
        lazy[i]=0;
    }
}
inline void update(int ll,int rr,int w,int l,int r,int i){
    if(ll<=l&&r<=rr){lazy[i]=(lazy[i]+w)%mod;sum[i]=(sum[i]+w*(r-l+1)%mod)%mod;return;}
    int mid((l+r)>>1);pushdown(i,r-l+1);
    if(ll<=mid)update(ll,rr,w,l,mid,i<<1);if(mid<rr)update(ll,rr,w,mid+1,r,i<<1|1);pushup(i);
}
inline int query(int ll,int rr,int l,int r,int i){
    if(ll<=l&&r<=rr)return sum[i];int ret(0),mid((l+r)>>1);pushdown(i,r-l+1);
    if(ll<=mid)ret=query(ll,rr,l,mid,i<<1);if(mid<rr)ret=(ret+query(ll,rr,mid+1,r,i<<1|1))%mod;return ret;
}
int ans[50005];
int fa[50005],son[50005],size[50005],dep[50005];
inline void dfs1(int u){
    son[u]=0;size[u]=1;
    for(edge *i=pre[u];i;i=i->n){
        int e(i->e);dep[e]=dep[u]+1;dfs1(e);
        size[u]+=size[e];if(size[e]>size[son[u]])son[u]=e;
    }
}
int cnt,id[50005],pos[50005],top[50005];
inline void dfs2(int u,int rt){
    top[u]=rt;id[u]=++cnt;pos[cnt]=u;if(son[u])dfs2(son[u],rt);
    for(edge *i=pre[u];i;i=i->n){int e(i->e);if(e==son[u])continue;dfs2(e,e);}
}
inline void change(int x,int y){
    while(top[x]^top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        update(id[top[x]],id[x],1,1,n,1);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    update(id[x],id[y],1,1,n,1);
}
inline int ask(int x,int y){
    int ret(0);
    while(top[x]^top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ret=(ret+query(id[top[x]],id[x],1,n,1))%mod;
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    ret=(ret+query(id[x],id[y],1,n,1))%mod;
    return ret;
}
int main(){
    n=read();q=read();
    for(int i=2;i<=n;++i){
        int x(read()+1);fa[i]=x;insert(x,i);
    }
    for(int i=1;i<=q;++i){
        int x(read()+1),y(read()+1);z[i]=read()+1;
        sub[x-1].push_back(i);add[y].push_back(i);
    }
    dfs1(1);dfs2(1,1);
    for(int i=1;i<=n;++i){
        change(i,1);
        for(int j=0,k=sub[i].size();j<k;++j)
            ans[sub[i][j]]=((ans[sub[i][j]]-ask(z[sub[i][j]],1))%mod+mod)%mod;
        for(int j=0,k=add[i].size();j<k;++j)
            ans[add[i][j]]=(ans[add[i][j]]+ask(z[add[i][j]],1))%mod;
    }
    for(int i=1;i<=q;++i)printf("%d\n",ans[i]);
}