DFS序+线段树(bzoj 4034)

题目链接

题目就不多说了。

本题目，可以用dfs序+线段树做：题目给定了一棵树，树上节点告诉了权值。我们可以先将这棵树进行dfs将一棵树变成线性结构：如图

变成这样后，然后就可以用线段树。

• 操作1：也就是将某两个点+a；
• 操作2：区间更新
• 操作3：查询起始区间到某点的和

我们建线段树，需要统计 +，- 抵消后的个数，因为要知道该区间的和，需要知道+a；

简单插线问线。

代码---参考下面链接吧或者我的

友情提示：注意爆int，计算的时候注意是否超int范围；所以wa了好多次。。。

 #include <cstdio>
 #include <cstring>
 #include <cctype>
 #include <cmath>
 #include <set>
 #include <map>
 #include <list>
 #include <queue>
 #include <deque>
 #include <stack>
 #include <string>
 #include <vector>
 #include <iostream>
 #include <algorithm>
 #include <stdlib.h>
 #include <time.h>
 using namespace std;
 typedef long long LL;
 const int INF=2e9+1e8;

 const int MOD=1e9+;
 const double eps=0.0000000001;
 void fre()
 {
     freopen("test.in","r",stdin);
     freopen("test.out","w",stdout);
 }
 #define MSET(a,b) memset(a,b,sizeof(a))

 const int maxn=1e6+;
 void zpsb(int x)
 {
     if(x>=&&x<maxn) return ;
     while();
 }
 struct Edge
 {
     int t,next;
 }edge[maxn];
 int sz,first[maxn],Treeval[maxn];
 void addedge(int s,int t)
 {
     edge[sz].t=t,edge[sz].next=first[s];
     first[s]=sz++;
 }
 int in[maxn],out[maxn];
 int reid[maxn],io[maxn],tot;
 void dfs(int x,int pre)
 {
     reid[in[x]=tot]=Treeval[x];
     io[tot++]=;
     for(int i=first[x];i!=-;i=edge[i].next)
     {
         int t=edge[i].t;
         if(t==pre) continue;
         dfs(t,x);
     }
     reid[out[x]=tot]=-Treeval[x];
     io[tot++]=-;
 }

 struct SegTree
 {
     struct Node
     {
         int l,r,flag;
         LL lazy,sum;
     }T[maxn*];
     void build(int i,int l,int r)
     {
         T[i].l=l,T[i].r=r;
         T[i].lazy=T[i].flag=;
         if(l==r)
         {
             T[i].sum=reid[r];
             T[i].flag=io[r];
             return ;
         }
         int mid=(l+r)>>;
         build(i<<,l,mid),build(i<<|,mid+,r);
         T[i].sum=T[i<<].sum+T[i<<|].sum;
         T[i].flag=T[i<<].flag+T[i<<|].flag;
     }
     void pushdown(int i)
     {
         if(T[i].lazy)
         {
             T[i<<].sum+=T[i].lazy*T[i<<].flag;
             T[i<<|].sum+=T[i].lazy*T[i<<|].flag;
             T[i<<].lazy+=T[i].lazy,T[i<<|].lazy+=T[i].lazy;
             T[i].lazy=;
         }
     }
     void update(int i,int l,int r,LL k)
     {
         zpsb(i);
         if(T[i].l==l&&T[i].r==r)
         {
             T[i].sum+=T[i].flag*k;
             T[i].lazy+=k;
             return ;
         }
         pushdown(i);
         int mid=(T[i].l+T[i].r)>>;
         if(r<=mid) update(i<<,l,r,k);
         else if(l>mid) update(i<<|,l,r,k);
         else update(i<<,l,mid,k),update(i<<|,mid+,r,k);
         T[i].sum=T[i<<].sum+T[i<<|].sum;
     }
     LL query(int i,int l,int r)
     {
         if(T[i].l==l&&T[i].r==r) return T[i].sum;
         pushdown(i);
         int mid=(T[i].l+T[i].r)>>;
         if(r<=mid) return query(i<<,l,r);
         else if(l>mid) return query(i<<|,l,r);
         else return query(i<<,l,mid)+query(i<<|,mid+,r);
     }
 }wa;
 int main()
 {
     int n,m;
     MSET(first,-);
     sz=;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++)
         scanf("%d",&Treeval[i]);
     for(int i=;i<=n;i++)
     {
         int x,y;
         scanf("%d%d",&x,&y);
         addedge(x,y);
         addedge(y,x);
     }
     tot=;
     dfs(,);
     // for(int i=1;i<=n;i++)
     // {
     //     printf("x=%d %d %d\n",i,in[i],out[i]);
     // }
     // for(int i=1;i<=2*n;i++)
     // {
     //     printf("id=%d val=%d\n",i,reid[i]);
     // }
     wa.build(,,*n);
     while(m--)
     {
         int opt;
         scanf("%d",&opt);
         if(opt==)
         {
             int x,a;
             scanf("%d%d",&x,&a);
             wa.update(,in[x],in[x],(LL)a);
             wa.update(,out[x],out[x],(LL)a);
         }
         else if(opt==)
         {
             int x,a;
             scanf("%d%d",&x,&a);
             wa.update(,in[x],out[x],(LL)a);
         }
         else
         {
             int x;
             scanf("%d",&x);
             printf("%lld\n",wa.query(,,in[x]));
         }
     }
 }

 /**************************************************/
 /**             Copyright Notice                 **/
 /**  writer: wurong                              **/
 /**  school: nyist                               **/
 /**  blog  : http://blog.csdn.net/wr_technology  **/
 /**************************************************/

参考资料：http://www.cnblogs.com/liyinggang/p/5965981.html