bzoj4034 （树链剖分+线段树）

Problem T2 （bzoj4034 HAOI2015）

题目大意

　　给定一颗树，1为根节点，要求支持三种操作。

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

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

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

解题分析

　　练手题。树链剖分+线段树。

参考程序

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

 #define V 100008
 #define E 200008
 #define lson l,m,rt << 1
 #define rson m+1,r,rt << 1 | 1
 #define LL long long

 struct line{
     int u,v,w,nt;
 }eg[E];
 int sum,lt[V];

 void adt(int u,int v){
     eg[++sum].u=u; eg[sum].v=v; eg[sum].nt=lt[u]; lt[u]=sum;
 }
 void add(int u,int v){
     adt(u,v); adt(v,u);
 }

 int size[V],fa[V],dep[V],son[V],w[V],top[V],rk[V];
 int n,m,a[V],cnt;

 void dfs_1(int u){
     size[u]=; dep[u]=dep[fa[u]]+; son[u]=;
     for (int i=lt[u];i;i=eg[i].nt){
         int v=eg[i].v;
         if (v==fa[u]) continue;
         fa[v]=u;
         dfs_1(v);
         size[u]+=size[v];
         if (size[v]>size[son[u]]) son[u]=v;
     }
 }

 void dfs_2(int u,int tp){
     top[u]=tp; w[u]=++cnt; rk[cnt]=u;
     if (son[u]) dfs_2(son[u],tp);
     for (int i=lt[u];i;i=eg[i].nt){
         int v=eg[i].v;
         if (v==fa[u] || v==son[u]) continue;
         dfs_2(v,v);
     }
 }

 struct segment_tree{
     LL sum[V << ],lazy[V << ];
     void pushup(int rt){
         sum[rt]=sum[rt<<]+sum[rt<<|];
     }
     void pushdown(int rt,int m){
         if (lazy[rt]){
             lazy[rt<<]+=lazy[rt];
             lazy[rt<<|]+=lazy[rt];
             sum[rt<<]+=lazy[rt] * (m-m/);
             sum[rt<<|]+=lazy[rt] * (m/);
             lazy[rt]=;
         }
     }
     void build(int l,int r,int rt){
         lazy[rt]=;
         if (l==r){
             sum[rt]=a[rk[l]];
             return;
         }
         int m=(l+r)/;
         build(lson);
         build(rson);
         pushup(rt);
     }
     void update(int L,int R,int val,int l,int r,int rt){
         if (L<=l && r<=R){
             lazy[rt]+=val;
             sum[rt]+=1ll*(r-l+)*val;
             return;
         }
         pushdown(rt,r-l+);
         int m=(l+r)/;
         if (L <= m) update(L,R,val,lson);
         if (m <  R) update(L,R,val,rson);
         pushup(rt);
     }
     LL query(int L,int R,int l,int r,int rt){
         if (L<=l && r<=R){
             return sum[rt];
         }
         pushdown(rt,r-l+);
         int m=(l+r)/;
         LL res=;
         if (L <= m) res+=query(L,R,lson);
         if (m <  R) res+=query(L,R,rson);
         return res;
     }
 }T;

 void change(int x,int y){
     LL res=;
     while (top[x]!=top[y]){
         if (dep[top[x]]<dep[top[y]]) swap(x,y);
         res+=T.query(w[top[x]],w[x],,n,);
         x=fa[top[x]];
     }
     if (dep[x]>dep[y]) swap(x,y);
     res+=T.query(w[x],w[y],,n,);
     printf("%lld\n",res );
 }

 int main(){
     memset(lt,,sizeof(lt)); sum=;
     scanf("%d %d",&n,&m);
     for (int i=;i<=n;i++) scanf("%d",&a[i]);
     for (int i=;i<n;i++){
         int u,v;
         scanf("%d %d",&u,&v);
         add(u,v);
     }
     dfs_1();
     dfs_2(,);
     T.build(,n,);
     while (m--){
         int x,y,z;
         scanf("%d",&x);
         if (x==){
             scanf("%d %d",&y,&z);
             T.update(w[y],w[y],z,,n,);
         }
         if (x==){
             scanf("%d %d",&y,&z);
             T.update(w[y],w[y]+size[y]-,z,,n,);
         }
         if (x==){
             scanf("%d",&y);
             change(,y);
         }
     }
 }