hdu5692 Snacks dfs序+线段树

题目传送门

题目大意：给出一颗树，根节点是0，有两种操作，一是修改某个节点的value，二是查询，从根节点出发，经过 x 节点的路径的最大值。

思路：用树状数组写发现还是有些麻烦，最后用线段树了。

　　　其实这道题的查询，就是查询从根节点到x节点+x节点走下去的路径的最大值，这样会发现，其实就是查询包括x节点的所有子树中权值最大的那个，而包括x节点的子树，如果用dfs序转换一下的话，可以在线段上用一段连续的点表示出来，所以最后就转换成了线段树区间求最大值，然后单点修改的题目了。

　　要注意的是dfs序和原标号的对应，有一个地方弄反了，卡了好久。

#pragma comment(linker, "/STACK:1024000000,1024000000") 
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<map>
#define CLR(a,b) memset(a,b,sizeof(a))
#define PI acos(-1)
#define lson rt*2,l,(l+r)/2
#define rson rt*2+1,(l+r)/2+1,r
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=100010;
struct edge{int to,Next;
}e[maxn*2];
int tot,m,n,head[maxn],pos,dfn[maxn],fa[maxn],son[maxn],l[maxn],r[maxn];
ll val[maxn],dis[maxn];
inline void init(){CLR(head,-1),tot=0,pos=0;CLR(dis,0);fa[1]=1;
}
inline void addv(int u,int v){e[++tot]={v,head[u]};head[u]=tot;
}
ll tree[maxn << 2], laz[maxn << 2];
inline void pushup(int rt) {tree[rt] = max(tree[rt << 1], tree[rt << 1 | 1]);
}
inline void pushdown(int rt) {if (laz[rt]) {tree[rt << 1] += laz[rt];tree[rt << 1 | 1] += laz[rt];laz[rt << 1] += laz[rt];laz[rt << 1 | 1] += laz[rt];laz[rt] = 0;}
}
inline void build(int rt, int l, int r) {laz[rt] = 0;if (l == r) {tree[rt] = dis[l];return ;}build(lson);build(rson);pushup(rt);
}
inline void update(int L, int R, ll v, int rt, int l, int r) {if (L <= l && R >= r) {tree[rt] += v;laz[rt] += v;return;}pushdown(rt);if (L <= (l + r) / 2)   update(L, R, v, lson);if (R > (l + r) / 2)    update(L, R, v, rson);pushup(rt);
}
inline ll query(int L, int R, int rt, int l, int r) { if (L <= l && R >= r) {return tree[rt];}pushdown(rt);if (L > (l + r) / 2)    return query(L,R,rson);else if (R <= (l + r) / 2)  return query(L,R,lson);else return max(query(L,R,lson),query(L,R,rson));
}inline void dfs(int u,int pre){dfn[u]=++pos;l[u]=pos;son[u]=1;dis[dfn[u]]=dis[dfn[pre]]+val[u];for(int i=head[u];i!=-1;i=e[i].Next){int v=e[i].to;if(v==fa[u])continue;fa[v]=u;dfs(v,u);son[u]+=son[v];}r[u]=pos;
}
int main(){int T,cas=1;cin>>T;while(T--){init();scanf("%d%d",&n,&m);for(int i=1;i<n;i++){int u,v;scanf("%d%d",&u,&v);addv(u+1,v+1);addv(v+1,u+1);}for(int i=1;i<=n;i++)scanf("%lld",&val[i]);dfs(1,1);build(1,1,n);printf("Case #%d:\n",cas++);while(m--){int op,x;ll y;scanf("%d%d",&op,&x);if(op==1){//    printf("debug  %d  %d\n",dfn[x+1],dfn[x+1]+son[x+1]-1);
            printf("%lld\n",query(l[x+1],r[x+1],1,1,n));}else{scanf("%lld",&y);update(l[x+1],r[x+1],y-val[x+1],1,1,n);val[x+1]=y;}}}
}