树剖想法题——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=;
 struct edge{LL to,nxt;}e[*N];
 struct node{LL s,t;}t[*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]=;son[k]=;
     for(LL i=head[k];i;i=e[i].nxt){
         LL v=e[i].to;
         if(v==fa)continue;
         dfs(v,k,dep+);
         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+);
     if(t[k].s>=mod)t[k].s%=mod;
     t[ls].t+=t[k].t; t[rs].t+=t[k].t;
     t[k].t=;
 }

 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+,R,l,r,x);
     pd(ls,L,mid);pd(rs,mid+,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 ;
     if(l<=L&&R<=r)return t[k].s;
     return qur(ls,L,mid,l,r)+qur(rs,mid+,R,l,r);
 }

 LL get(LL x){
     LL ret=;
     while(x){
         ret+=qur(,,cnt,id[top[x]],id[x]);
         if(ret>=mod)ret%=mod;
         x=f[top[x]];
     }
     return ret;
 }

 void change(LL x){
     while(x){
         update(,,cnt,id[top[x]],id[x],);
         x=f[top[x]];
     }
 }
 int main(){
     LL x,l,r;
     scanf("%d%d",&n,&m);
     foru(i,,n){
         scanf("%d",&x);x++;
         add(x,i);add(i,x);
     }
     foru(i,,m){
         scanf("%d%d%d",&l,&r,&q[i]);q[i]++;
         a[++nq].p=l;a[nq].id=i;a[nq].flag=-;
         a[++nq].p=r+;a[nq].id=i;a[nq].flag=;
     }
     dfs(,,);
     build(,);
     sort(a+,a++nq,cmp);
     LL j=;
     while(!a[j+].p)j++;
     foru(i,,n){
         change(i);
         while(a[j+].p==i)j++,ans[a[j].id]+=get(q[a[j].id])*a[j].flag;
     }
     foru(i,,m)printf("%d\n",(ans[i]+mod)%mod);
     return ;
 }