HDU 5692 Snacks DFS序 + 线段树区间更新

题目链接： http://acm.hdu.edu.cn/showproblem.php?pid=5692
题意：问一个子树中，到树的根节点路径上节点最大和是多少

这样可以用DFS序来维护这个树结构， 将一棵子树映射到一段区间类，区间每个树存的是相应节点到根的路径节点和，那么显然可以用一个线段树来维护这个区间的最大值。
在更新节点值的时候，我们只需要将这个节点为根的子树映射到的区间更新即可

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000") `
#include <bits/stdc++.h>
#define sf scanf
#define pf printf
typedef long long LL;
using namespace std;
const int maxn = 100000 + 50;
const LL INF = 1e18;
struct Edge{
    int v,pre;
}Es[2 * maxn];
int head[maxn],tot;
void EDGE_INIT(){
    memset(head,-1,sizeof head),tot = 0;
}
void ADD_EDGE(int u,int v){
    Es[tot].v = v;
    Es[tot].pre = head[u];
    head[u] = tot++;
}

/** SegTree */
LL _MAX[maxn << 2],_LAZE[maxn << 2],_AR[maxn];
#define lson rt << 1 , l , mid
#define rson rt << 1 | 1,mid + 1,r
void PushUp(int rt){
    _MAX[rt] = max(_MAX[rt << 1] , _MAX[rt << 1 | 1]);
}
void PushDown(int rt){
    if(_LAZE[rt]){
        _LAZE[rt << 1] += _LAZE[rt];
        _LAZE[rt << 1 | 1] += _LAZE[rt];

        _MAX[rt << 1] += _LAZE[rt];
        _MAX[rt << 1 | 1] += _LAZE[rt];

        _LAZE[rt] = 0;
    }
}
void Build(int rt,int l,int r){
    if(l == r){
        _LAZE[rt] = 0;
        _MAX[rt] = _AR[l];return;
    }
    int mid = l + r >> 1;
    Build(lson),Build(rson);
    _LAZE[rt] = 0;PushUp(rt);
}
void Update(int rt,int l,int r,int L,int R,int x){
    if(L <= l && R >= r){
        _MAX[rt] += x;
        _LAZE[rt] += x;
        return;
    }
    PushDown(rt);
    int mid = l + r >> 1;
    if(L <= mid) Update(lson,L,R,x);
    if(R > mid) Update(rson,L,R,x);
    PushUp(rt);
}
LL Query(int rt,int l,int r,int L,int R){
    if(L <= l && R >= r){
        return _MAX[rt];
    }
    PushDown(rt);
    int mid = l + r >> 1;
    LL ret = -INF;
    if(L <= mid) ret = max(ret,Query(lson,L,R));
    if(R > mid) ret = max(ret,Query(rson,L,R));
    PushUp(rt);
    return ret;
}
/** SegTree END */

/** DFS_SEQUENCE */
int L[maxn],R[maxn];
LL A[maxn];
int _TOT_TIME;
void DFS(int u,int fa){
    L[u] = ++_TOT_TIME;
    _AR[L[u]] = _AR[L[fa]] + A[u];
    for(int i = head[u];~i;i = Es[i].pre){
        int v = Es[i].v;
        if(v == fa) continue;
        DFS(v,u);
    }
    R[u] = _TOT_TIME;
}
void DFS_INIT(){
    _TOT_TIME = 0;
    _AR[0] = 0;L[0] = 0;
    DFS(1,0);
}

int main(){
    int T,ca = 0;sf("%d",&T);
    while( T-- ){
        int n,m;sf("%d %d",&n,&m);
        EDGE_INIT();
        for(int i = 1;i < n;++i){
            int u,v;sf("%d %d",&u,&v);u++,v++;
            ADD_EDGE(u,v);
            ADD_EDGE(v,u);
        }
        for(int i = 1;i <= n;++i) sf("%lld",&A[i]);
        DFS_INIT();
        Build(1,1,n);
        pf("Case #%d:\n",++ca);
        while(m--){
            int op,x,y;
            sf("%d",&op);
            if(op == 0){
                sf("%d %d",&x,&y);x++;
                Update(1,1,n,L[x],R[x],y - A[x]);
                A[x] = y;
            }
            else{
                sf("%d",&x);x++;
                pf("%lld\n",Query(1,1,n,L[x],R[x]));
            }
        }
    }
}