洛谷 3178 [HAOI2015]树上操作

【题解】

　　就是个树链剖分的模板题。

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define LL long long
 #define rg register
 #define N 200010
 #define ls (u<<1)
 #define rs (u<<1|1)
 #define mid ((a[u].l+a[u].r)>>1)
 using namespace std;
 int n,m,tot,cnt,opt,last[N],fa[N],hvy[N],top[N],dep[N],dfn[N],siz[N],poi[N],v[N];
 struct edge{
     int to,pre;
 }e[N<<];
 struct tree{
     int l,r; LL sum,del;
 }a[N<<];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     return k*f;
 }
 void dfs1(int x){
     siz[x]=; dep[x]=dep[fa[x]]+;
     for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa[x]){
         fa[to]=x; dfs1(to); siz[x]+=siz[to];
         if(siz[to]>siz[hvy[x]]) hvy[x]=to;
     }
 }
 void dfs2(int x,int tp){
     top[x]=tp; dfn[x]=++cnt; poi[cnt]=x;
     if(hvy[x]) dfs2(hvy[x],tp);
     for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa[x]&&to!=hvy[x])
     dfs2(to,to);
 }
 void build(int u,int l,int r){
     a[u].l=l; a[u].r=r;
     if(l<r) build(ls,l,mid),build(rs,mid+,r),a[u].sum=a[ls].sum+a[rs].sum;
     else a[u].sum=v[poi[l]];
 }
 inline void pushdown(int u){
     LL d=a[u].del; a[u].del=;
     a[ls].sum+=1LL*d*(a[ls].r-a[ls].l+); a[ls].del+=d;
     a[rs].sum+=1LL*d*(a[rs].r-a[rs].l+); a[rs].del+=d;
 }
 void update(int u,int l,int r,LL del){
     if(l<=a[u].l&&a[u].r<=r){
         a[u].sum+=1LL*del*(a[u].r-a[u].l+);
         a[u].del+=del;
         return;
     }
     if(a[u].del) pushdown(u);
     if(l<=mid) update(ls,l,r,del);
     if(r>mid) update(rs,l,r,del);
     a[u].sum=a[ls].sum+a[rs].sum;
 }
 LL query(int u,int l,int r){
     if(l<=a[u].l&&a[u].r<=r) return a[u].sum;
     if(a[u].del) pushdown(u); LL ret=;
     if(l<=mid) ret+=query(ls,l,r);
     if(r>mid) ret+=query(rs,l,r);
     return ret;
 }
 int main(){
     n=read(); m=read();
     for(rg int i=;i<=n;i++) v[i]=read();
     for(rg int i=;i<n;i++){
         int u=read(),v=read();
         e[++tot]=(edge){v,last[u]}; last[u]=tot;
         e[++tot]=(edge){u,last[v]}; last[v]=tot;
     }
     dfs1(); dfs2(,); build(,,n);
     while(m--){
 //        printf("[--> %lld]",query(1,dfn[2],dfn[2]+1));
         opt=read();
         if(opt==){
             int x=read(),y=read();
             update(,dfn[x],dfn[x],y);
         }
         else{
             if(opt==){
                 int x=read(),y=read();
                 update(,dfn[x],dfn[x]+siz[x]-,y);
             }
             else{
                 int x=read(),t=top[x]; LL ret=;
                 while(x){
                     ret+=query(,dfn[t],dfn[x]);
                     x=fa[t]; t=top[x];
 //                    printf("[%d]\n",x);
                 }
                 printf("%lld\n",ret);
             }
         }
     }
     return ;
 }