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

来源：互联网发布：pc群控手机源码编辑：程序博客网时间：2024/05/17 08:24

题目链接： 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 printftypedef 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,rvoid 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]));            }        }    }}

0 0