hdu5692 Snacks(线段树+dfs序)

最新推荐文章于 2019-09-12 10:40:07 发布

原创最新推荐文章于 2019-09-12 10:40:07 发布 · 446 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

其他oj 同时被 3 个专栏收录

336 篇文章

订阅专栏

线段树

123 篇文章

订阅专栏

-----树-------

104 篇文章

订阅专栏

本文介绍了一种使用线段树优化树形结构的方法，通过维护从每个节点到根节点的路径权值和，实现了对特定子树中节点到根节点距离的最大值的高效查询与更新。该方法特别适用于需要频繁进行此类操作的问题。

我们维护每个点到根的权值和。询问过x的路径中权值最大的，即询问x的子树中到根的距离最大的。修改x为y，则x的子树中的所有点到根的距离都加上了y-a[x]。显然可以用dfn变成连续区间用线段树来维护。

#pragma comment(linker, "/STACK:1024000000,1024000000") 
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 1LL<<63
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int tst,n,m,h[N],num=0,in[N],out[N],a[N],dfn,id[N];
ll d[N];
struct edge{
    int to,next;
}data[N<<1];
struct node{
    ll mx,add;
}tree[N<<2];
void dfs(int x){
    in[x]=++dfn;id[dfn]=x;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(in[y]) continue;
        d[y]=d[x]+a[y];dfs(y);
    }
    out[x]=dfn;
}
inline void pushup(int p){
    tree[p].mx=max(tree[p<<1].mx,tree[p<<1|1].mx);
}
inline void pushdown(int p){
    if(!tree[p].add) return;
    tree[p<<1].add+=tree[p].add;tree[p<<1|1].add+=tree[p].add;
    tree[p<<1].mx+=tree[p].add;tree[p<<1|1].mx+=tree[p].add;
    tree[p].add=0;
}
void build(int p,int l,int r){
    tree[p].add=0;
    if(l==r){tree[p].mx=d[id[l]];return;}
    int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
    pushup(p);
}
void add(int p,int l,int r,int x,int y,int val){
    if(x<=l&&r<=y){tree[p].mx+=val;tree[p].add+=val;return;}
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) add(p<<1,l,mid,x,y,val);
    if(y>mid) add(p<<1|1,mid+1,r,x,y,val);
    pushup(p);
}
ll qmax(int p,int l,int r,int x,int y){
    if(x<=l&&r<=y) return tree[p].mx;
    int mid=l+r>>1;pushdown(p);ll res=-inf;
    if(x<=mid) res=max(res,qmax(p<<1,l,mid,x,y));
    if(y>mid) res=max(res,qmax(p<<1|1,mid+1,r,x,y));
    return res;
}
int main(){
//  freopen("a.in","r",stdin);
    tst=read();
    for(int tt=1;tt<=tst;++tt){
        printf("Case #%d:\n",tt);
        n=read();m=read();memset(h,0,sizeof(h));num=0;
        dfn=0;memset(in,0,sizeof(in));
        for(int i=1;i<n;++i){
            int x=read(),y=read();
            data[++num].to=y;data[num].next=h[x];h[x]=num;
            data[++num].to=x;data[num].next=h[y];h[y]=num;
        }
        for(int i=0;i<n;++i) a[i]=read();
        d[0]=a[0];dfs(0);build(1,1,n);
        while(m--){
            int op=read(),x=read();
            if(op) printf("%lld\n",qmax(1,1,n,in[x],out[x]));
            else{
                int y=read();
                add(1,1,n,in[x],out[x],y-a[x]);a[x]=y;
            }
        }
    }
    return 0;
}