hdu 5692(dfs+线段树) Snacks

题目http://acm.hdu.edu.cn/showproblem.php?pid=5692

题目说每个点至多经过一次，那么就是只能一条路线走到底的意思，看到这题的格式，

多个询问多个更新，

自然而然的就会想到线段树或者树状数组，在建树前先做处理，

用DFS将从起点0出发到任一点的距离求出，

然后将这些节点按照一条一条完整的路线的顺序建到树中，

比如样例是1---2---3

|

6 ---4----5

所以建树的其中一种顺序是1 4 5 6 2 3 。当查询的时候的区间应该是从现在这个点开始到这条路线上的终点，更新的时候也是这个区间更新，相当于把这个区间的所有数都加上一个数，与 poj 3468 类似，lazy操作。

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

 typedef long long ll;
 const int M = 1e5 + ;
 int head[M],cas,num[M],ans,st[M],ed[M];
 ll sum[M],a[M];

 ll max(ll x,ll y) {return x>y?x:y;}

 struct Edge{
    int to;
    int next;
 }edge[M*];

 void add(int x,int y)
 {
     edge[cas].to=x;
     edge[cas].next=head[y];
     head[y]=cas++;
 }

 void dfs(int x,int y)
 {
     st[x]=++ans;num[ans]=x;
     sum[x]+=sum[y];
     for (int i=head[x] ; i!=- ; i=edge[i].next)
     {
         int v=edge[i].to;
         if (v==y) continue;
         dfs(v,x);
     }
     ed[x]=ans;
 }
 struct Tree{
     int l,r;
     ll ans,mark;
 }tree[M*];

 void down(int i)
 {
     if(tree[i].mark)
     {
         tree[i*].mark+=tree[i].mark;tree[i*+].mark+=tree[i].mark;
         tree[i*].ans+=tree[i].mark;tree[i*+].ans+=tree[i].mark;
         tree[i].mark=;
     }
 }

 void build(int i,int left,int right)
 {
     tree[i].l=left;tree[i].r=right;
     tree[i].mark=;
     if (left==right){tree[i].ans=sum[num[left]];return ;}
     int mid=(left+right)/;
     build(i*,left,mid);
     build(i*+,mid+,right);
     tree[i].ans=max(tree[i*].ans,tree[i*+].ans);
 }

 ll findest(int i,int left,int right)
 {
     if(tree[i].l>=left&&tree[i].r<=right)
         return tree[i].ans;
     down(i);
     int mid=(tree[i].l+tree[i].r)/;
     if(mid>=right){return findest(i*,left,right);}
     else if(mid<left){return findest(i*+,left,right);}
     else{return max(findest(i*,left,mid),findest(i*+,mid+,right));}

 }
 void update(int i,int left,int right,ll pos)
 {
     if(tree[i].l>=left&&tree[i].r<=right){tree[i].mark+=pos;tree[i].ans+=pos;return;}
     int mid=(tree[i].l+tree[i].r)/;
     down(i);
     if(mid<left) update(i*+,left,right,pos);
     else if(mid>=right) update(i*,left,right,pos);
     else
     {
         update(i*,left,mid,pos);
         update(i*+,mid+,right,pos);
     }
     tree[i].ans=max(tree[i*].ans,tree[i*+].ans);
 }

 int main()
 {
     int t,n,m,e=;
     scanf("%d",&t);
     while (t--)
     {
         printf("Case #%d:\n",++e);
         scanf("%d%d",&n,&m);
         memset(head,-,sizeof(head));
         for (int i= ; i<n ; i++){
             int a,b;
             scanf("%d%d",&a,&b);
             a++,b++;
             add(a,b);add(b,a);
         }
         cas=;ans=;
         for (int i= ; i<=n ; i++) scanf("%I64d",&a[i]),sum[i]=a[i];
         dfs(,);
         build(,,n);
         while (m--)
         {
             int w,b;ll c;
             scanf("%d",&w);
             if (w){
                 scanf("%d",&b);b++;
                 printf("%I64d\n",findest(,st[b],ed[b]));
             }
             else{
                 scanf("%d%I64d",&b,&c);b++;
                 ll q=c-a[b];
                 a[b]=c;
                 update(,st[b],ed[b],q);
             }
         }
     }
     return ;
 }