【bzoj4034】[HAOI2015]T2

siz[v]表示以v为根的子树的节点数

top[v]表示v所在的重链的顶端节点

fa[v]表示v的父亲

pos[v]表示v的父边标号

mx[v]表示v的子树中边的标号最大的那条边

参考:http://blog.sina.com.cn/s/blog_6974c8b20100zc61.html

题意：

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个

操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

第一次写树链剖分。。

 #include<algorithm>
 #include<iostream>
 #include<cstdlib>
 #include<cstring>
 #include<cstdio>
 #include<cmath>
 using namespace std;

 typedef long long LL;

 #define N 100010

 int id;
 int fa[N],siz[N],top[N];
 int head[N],pos[N],mx[N],v[N];
 LL sum[N<<],add[N<<];

 struct Node
 {
     int to,next;
 }e[N<<];
 int cnt;

 int n,m;
 int uu,vv;

 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 void link(int x,int y)
 {
     e[++cnt]=(Node){y,head[x]};
     head[x]=cnt;
 }

 void dfs(int x)
 {
     siz[x]=;
     for (int i=head[x];i;i=e[i].next)
         if (e[i].to!=fa[x])
         {
             fa[e[i].to]=x;
             dfs(e[i].to);
             siz[x]+=siz[e[i].to];
             mx[x]=max(mx[x],mx[e[i].to]);
         }
 }

 void dfs2(int x,int cha)
 {
     top[x]=cha;pos[x]=mx[x]=++id;
     int k=;
     for(int i=head[x];i;i=e[i].next)
         if(e[i].to!=fa[x]&&siz[e[i].to]>siz[k])
             k=e[i].to;
     if(k)
     {
         dfs2(k,cha);mx[x]=max(mx[x],mx[k]);
     }
     for(int i=head[x];i;i=e[i].next)
         if(e[i].to!=fa[x]&&e[i].to!=k)
         {
             dfs2(e[i].to,e[i].to);
             mx[x]=max(mx[x],mx[e[i].to]);
         }
 }

 void pushup(int now)
 {
     sum[now]=sum[now<<]+sum[now<<|];
 }

 void pushdown(int nowl,int nowr,int now)
 {
     if (nowl==nowr)
         return;
     int mid=(nowl+nowr)>>;
     LL t=add[now];
     add[now]=;
     add[now<<]+=t;
     add[now<<|]+=t;
     sum[now<<]+=t*(mid-nowl+);
     sum[now<<|]+=t*(nowr-mid);
 }

 void update(int nowl,int nowr,int now,int l,int r,LL d)
 {
     if (add[now])
         pushdown(nowl,nowr,now);
     if (nowl==l && nowr==r)
     {
         add[now]+=d;
         sum[now]+=(nowr-nowl+)*d;
         return ;
     }
     int mid=(nowl+nowr)>>;
     if (l<=mid)
         update(nowl,mid,now<<,l,min(r,mid),d);
     if (r>mid)
         update(mid+,nowr,now<<|,max(l,mid+),r,d);
     pushup(now);
 }

 LL query(int nowl,int nowr,int now,int l,int r)
 {
     if (add[now])
         pushdown(nowl,nowr,now);
     if (nowl==l && nowr==r)
         return sum[now];
     int mid=(nowl+nowr)>>;
     LL ans=;
     if (l<=mid)
         ans+=query(nowl,mid,now<<,l,min(mid,r));
     if (r>mid)
         ans+=query(mid+,nowr,now<<|,max(mid+,l),r);
     return ans;
 }

 LL query(int x)
 {
     LL ans=;
     while (top[x]!=)
     {
         ans+=query(,n,,pos[top[x]],pos[x]);
         x=fa[top[x]];
     }
     ans+=query(,n,,,pos[x]);
     return ans;
 }

 int main()
 {
     n=read(),m=read();
     for (int i=;i<=n;i++)
         v[i]=read();
     for (int i=;i<n;i++)
     {
         uu=read(),vv=read();
         link(uu,vv);
         link(vv,uu);
     }
     dfs();
     dfs2(,);
     for (int i=;i<=n;i++)
         update(,n,,pos[i],pos[i],v[i]);
     int askd,x,a;
     while (m--)
     {
         askd=read(),x=read();
         if (askd==)
         {
             a=read();
             update(,n,,pos[x],pos[x],a);
         }
         if (askd==)
         {
             a=read();
             update(,n,,pos[x],mx[x],a);
         }
         if (askd==)
             printf("%lld\n",query(x));
     }
     return ;
 }