2016"百度之星" - 初赛（Astar Round2A） 1003 【dfs序+线段树】

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5692

分析：每次询问在x的子树里找里root最远的点，很容易想到用dfs序找到他子树的区间，直接用线段树求出就可以了，更新的时候只要在子树的每个节点上加上修改的差值。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define Mn 2000010
#define Mm 2000005
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
ll INF=1LL<<62;
int read() {
    char c;
    int ans=0,f=1;c=getchar();
    while(c<'0'||c>'9') {if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9') {ans=ans*10+c-'0';c=getchar();}
    return  ans*f;
}
struct edge {
    int v,next;
}e[Mm];
int tot,head[Mn];
void addedge(int v,int u) {
    e[tot].v=v;
    e[tot].next=head[u];
    head[u]=tot++;
}
int dfs_clock;
int ldfn[Mn],rdfn[Mn];
ll dis[Mn];
int w[Mn];
int id[Mn];
void dfs(int u,int fa) {
    ldfn[u]=++dfs_clock;
    dis[u]=dis[fa]+w[u];
    id[dfs_clock]=u;
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        if(v==fa) continue;
        dfs(v,u);
    }
    rdfn[u]=dfs_clock;
}
ll maxx[Mn];
ll lazy[Mn];
void build(int l,int r,int u) {
    lazy[u]=0;
    if(l==r) {
        maxx[u]=dis[id[l]];
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,ul);
    build(mid+1,r,ur);
    maxx[u]=max(maxx[ul],maxx[ur]);
}
int s,t;
void push_down(int u) {
    lazy[ul]+=lazy[u];
    lazy[ur]+=lazy[u];
    maxx[ul]+=lazy[u];
    maxx[ur]+=lazy[u];
    lazy[u]=0;
}
ll query(int l,int r,int u) {
    if(s<=l&&t>=r) {
        return maxx[u];
    }
    if(lazy[u]) push_down(u);
    int mid=(l+r)>>1;
    ll re1=-INF,re2=re1;
    if(s<=mid) re1=query(l,mid,ul);
    if(t>mid) re2=query(mid+1,r,ur);
    return max(re1,re2);
}
void updata(int l,int r,int u,int val) {
    if(s<=l&&t>=r) {
        maxx[u]+=(ll)val;
        lazy[u]+=(ll)val;
        return ;
    }
    if(lazy[u]) push_down(u);
    int mid=(l+r)>>1;
    if(s<=mid) updata(l,mid,ul,val);
    if(t>mid) updata(mid+1,r,ur,val);
    maxx[u]=max(maxx[ul],maxx[ur]);
}
void init() {
    CLR(head,-1);
    tot=0;
    dfs_clock=0;
}
int main() {
    int T,cas=0;
    int n,m;
    T=read();
    while(T--) {
        n=read(),m=read();
        int u,v;init();
        for(int i=1;i<n;i++) {
            u=read(),v=read();
            u++,v++;
            addedge(u,v);
            addedge(v,u);
        }
        for(int i=1;i<=n;i++) {
            w[i]=read();
        }
        dfs(1,0);
        build(1,n,1);
        printf("Case #%d:\n",++cas);
        for(int i=1;i<=m;i++) {
            int x,y,z;
            x=read();y=read()+1;
            if(x==1) {
                s=ldfn[y],t=rdfn[y];
                printf("%I64d\n",query(1,n,1));
            } else {
                z=read();
                s=ldfn[y],t=rdfn[y];
                updata(1,n,1,z-w[y]);
                w[y]=z;
            }
        }

    }
    return 0;
}