BZOJ 3626: [LNOI2014]LCA [树链剖分 离线|主席树]

//
//  main.cpp
//  bzoj4196
//
//  Created by Candy on 2017/1/2.
//  Copyright © 2017年 Candy. All rights reserved.
//

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define lc o<<1
#define rc o<<1|1
#define m ((l+r)>>1)
#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
const int N=1e5+5,MOD=201314;
typedef long long ll;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<‘0‘||c>‘9‘){if(c==‘-‘)f=-1; c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){x=x*10+c-‘0‘; c=getchar();}
    return x*f;
}
int n,Q,u,l,r;
char s[20];
struct ques{
    int z,al,ar;
}q[N];
struct que{
    int x,id,isl;
    bool operator <(const que &r)const{return x<r.x;}
}a[N];
int p;
struct edge{
    int v,ne,c,f;
}e[N<<1];
int cnt,h[N];
inline void ins(int u,int v){
    cnt++;
    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
}

int deep[N],fa[N],tid[N],tot,top[N],mx[N],size[N];
void dfs(int u){
    size[u]=1;
    for(int i=h[u];i;i=e[i].ne){
        int v=e[i].v;
        if(v==fa[u]) continue;
        fa[v]=u;deep[v]=deep[u]+1;
        dfs(v);
        size[u]+=size[v];
        if(size[v]>size[mx[u]]) mx[u]=v;
    }
}
void dfs(int u,int anc){
    if(!u) return;
    tid[u]=++tot;
    top[u]=anc;
    dfs(mx[u],anc);
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].v!=mx[u]&&e[i].v!=fa[u]) dfs(e[i].v,e[i].v);
}

struct node{
    int sum,add;
}t[N<<2];
inline void merge(int o){
    t[o].sum=t[lc].sum+t[rc].sum;
}
inline void paint(int o,int l,int r,int d){
    t[o].sum+=d*(r-l+1);
    t[o].add+=d;
}
inline void pushDown(int o,int l,int r){
    if(t[o].add){
        paint(lson,t[o].add);
        paint(rson,t[o].add);
        t[o].add=0;
    }
}
inline void segAdd(int o,int l,int r,int ql,int qr,int d){//printf("add %d %d %d\n",o,l,r);
    if(ql<=l&&r<=qr) paint(o,l,r,d);
    else{
        pushDown(o,l,r);
        if(ql<=m) segAdd(lson,ql,qr,d);
        if(m<qr) segAdd(rson,ql,qr,d);
        merge(o);
    }
}
inline int segQue(int o,int l,int r,int ql,int qr){//printf("que %d %d %d %d %d\n",o,l,r,ql,qr);
    if(ql<=l&&r<=qr) return t[o].sum;
    else{
        pushDown(o,l,r);
        int ans=0;
        if(ql<=m) ans+=segQue(lson,ql,qr);
        if(m<qr) ans+=segQue(rson,ql,qr);
        return ans;
    }
}
void build(int o,int l,int r){
    if(l==r) paint(o,l,r,0);
    else{
        build(lson);
        build(rson);
    }
}

void add(int x,int y,int d){
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        segAdd(1,1,n,tid[top[x]],tid[x],d);
        x=fa[top[x]];
    }
    if(tid[x]>tid[y]) swap(x,y);
    segAdd(1,1,n,tid[x],tid[y],d);
}
int query(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        ans+=segQue(1,1,n,tid[top[x]],tid[x]);ans%=MOD;
        x=fa[top[x]];
    }
    if(tid[x]>tid[y]) swap(x,y);
    ans+=segQue(1,1,n,tid[x],tid[y]);ans%=MOD;
    return ans;
}
int main(){
    n=read();Q=read();
    for(int i=2;i<=n;i++) u=read()+1,ins(u,i);
    dfs(1);dfs(1,1);
    build(1,1,n);
    for(int i=1;i<=Q;i++){
        l=read();r=read()+1;q[i].z=read()+1;
        a[++p].x=l;a[p].id=i;a[p].isl=1;
        a[++p].x=r;a[p].id=i;a[p].isl=0;
    }
    sort(a+1,a+1+p);
    int now=1;
    for(int i=1;i<=p;i++){//printf("a %d %d %d\n",a[i].x,a[i].id,a[i].isl);
        while(now<=a[i].x) add(1,now,1),now++;
        int _=a[i].id;
        if(a[i].isl) q[_].al=query(1,q[_].z);
        else q[_].ar=query(1,q[_].z);
    }
    for(int i=1;i<=Q;i++) printf("%d\n",(q[i].ar-q[i].al+MOD)%MOD);
}