树剖想法题——BZOJ3626

本来是打算作为树剖练习的最后一题的，结果一直WA。

本来以为是自己写的太丑。

最后发现5w的数据

我开了10w的数组

然而有一个数组要×2

哦，好棒棒。

#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define foru(i,x,y) for(LL i=x;i<=y;i++)
using namespace std;
typedef int LL;
const LL N=2e5;
const LL mod=201314;
struct edge{LL to,nxt;}e[2*N];
struct node{LL s,t;}t[10*N];
struct que{LL id,flag,p;}a[N];
LL f[N],b[N],d[N],id[N],ans[N],siz[N],son[N],top[N],head[N],
   cnt,ne,n,m,nq,q[N];
    
bool cmp(que a,que b){return a.p<b.p;}
    
void add(LL a,LL b){
    e[++ne]=(edge){b,head[a]};head[a]=ne;
}
    
void dfs(LL k,LL fa,LL dep){
    f[k]=fa;d[k]=dep;siz[k]=1;son[k]=0;
    for(LL i=head[k];i;i=e[i].nxt){
        LL v=e[i].to;
        if(v==fa)continue;
        dfs(v,k,dep+1);
        siz[k]+=siz[v];
        if(siz[v]>siz[son[k]])son[k]=v;
    }
}
    
void build(LL k,LL tp){
    id[k]=++cnt;top[k]=tp;
    if(son[k])build(son[k],tp);
    for(LL i=head[k];i;i=e[i].nxt){
        LL v=e[i].to;
        if(v!=son[k]&&v!=f[k])build(v,v);
    }
}
    
#define mid ((L+R)>>1)
#define ls (k<<1)
#define rs ls+1
    
void pd(LL k,LL L,LL R){
    t[k].s+=t[k].t*(R-L+1);
    if(t[k].s>=mod)t[k].s%=mod;
    t[ls].t+=t[k].t; t[rs].t+=t[k].t;
    t[k].t=0;
}
    
void pu(LL k,LL L,LL R){
    t[k].s=t[ls].s+t[rs].s;
    if(t[k].s>=mod)t[k].s%=mod;
}
    
void update(LL k,LL L,LL R,LL l,LL r,LL x){
    pd(k,L,R);
    if(r<L||l>R)return;
    if(l<=L&&R<=r){t[k].t+=x;return;}
    update(ls,L,mid,l,r,x); update(rs,mid+1,R,l,r,x);
    pd(ls,L,mid);pd(rs,mid+1,R);
    pu(k,L,R);
}
    
LL qur(LL k,LL L,LL R,LL l,LL r){
    pd(k,L,R);
    if(l>R||r<L)return 0;
    if(l<=L&&R<=r)return t[k].s;
    return qur(ls,L,mid,l,r)+qur(rs,mid+1,R,l,r);
}
    
LL get(LL x){
    LL ret=0;
    while(x){
        ret+=qur(1,1,cnt,id[top[x]],id[x]);
        if(ret>=mod)ret%=mod;
        x=f[top[x]];
    }
    return ret;
}
    
void change(LL x){
    while(x){
        update(1,1,cnt,id[top[x]],id[x],1);
        x=f[top[x]];
    }
}
int main(){
    LL x,l,r;
    scanf("%d%d",&n,&m);
    foru(i,2,n){
        scanf("%d",&x);x++;
        add(x,i);add(i,x);
    }
    foru(i,1,m){
        scanf("%d%d%d",&l,&r,&q[i]);q[i]++;
        a[++nq].p=l;a[nq].id=i;a[nq].flag=-1;
        a[++nq].p=r+1;a[nq].id=i;a[nq].flag=1;
    }
    dfs(1,0,1);
    build(1,1);
    sort(a+1,a+1+nq,cmp);
    LL j=0;
    while(!a[j+1].p)j++;
    foru(i,1,n){
        change(i);
        while(a[j+1].p==i)j++,ans[a[j].id]+=get(q[a[j].id])*a[j].flag;
    }
    foru(i,1,m)printf("%d\n",(ans[i]+mod)%mod);
    return 0;
}