hdu5692 Snacks dfs序+线段树

Posted 2021-01-15 mountaink

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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:
",cas++);
        while(m--){
            int op,x;ll y;
            scanf("%d%d",&op,&x);
            if(op==1){
            //    printf("debug  %d  %d
",dfn[x+1],dfn[x+1]+son[x+1]-1);
            
                printf("%lld
",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;
            }
        }
        
    }
}