codevs 2460 [ZJOI] 树的统计 树剖

题目：
—–

Q

第一遍dfs找重边；
第二遍dfs连重链；
然后线段树维护dfs序；

—————————————————————————————
以下来自loi_DQS学长的博客;

deep[u]:u点深度
sz[u]：u的子树大小(儿子数，包括自己);
son[u]:u的重儿子（重儿子定义：儿子v中sz[v]最大的儿子）
fa[u]:u的爹
inseg[u]：树中的点u在线段树中的标号。
intr[totp]:线段树的点totp在树中的标号。
top[u]：u所在重链的顶端节点,即起点。

——————————————————————————————

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=140001;
const int inf=2147483647;
int fst[MAXN],nxt[MAXN],deep[MAXN],fa[MAXN],inseg[MAXN],intr[MAXN];
int son[MAXN],sz[MAXN],num[MAXN],top[MAXN];
struct hh {int from,to;}ma[MAXN];
struct xx {int maxx,l,r,sum;}tree[MAXN];
int tot,totp,n;
void build(int f,int t)
{
    tot++;
    ma[tot]=(hh){f,t};
    nxt[tot]=fst[f];
    fst[f]=tot;
    return;
}
void dfs1(int u,int f)
{
    fa[u]=f;
    sz[u]=1;
    deep[u]=deep[f]+1;
    for(int i=fst[u];i;i=nxt[i])
    {
        int v=ma[i].to;
        if(v==f) continue;
        dfs1(v,u);
        sz[u]+=sz[v];
        if(!son[u] || sz[v]>sz[son[u]]) son[u]=v;
    }
    return;
}
void dfs2(int u,int st)
{
    top[u]=st;
    inseg[u]=++totp;
    intr[totp]=u;
    if(!son[u]) return;
    dfs2(son[u],st);
    for(int i=fst[u];i;i=nxt[i])
    {
        int v=ma[i].to;
        if(v==fa[u] || v==son[u]) continue;
        dfs2(v,v);
    }
    return;
}
void up(int now)
{
    tree[now].sum=tree[now<<1].sum+tree[now<<1|1].sum;
    tree[now].maxx=max(tree[now<<1].maxx,tree[now<<1|1].maxx);
    return;
}
void add(int now,int l,int r)
{
    tree[now].l=l;
    tree[now].r=r;
    if(l==r)
    {
        tree[now].sum=tree[now].maxx=num[intr[l]];
        return;
    }
    int mid=(l+r)>>1;
    add(now<<1,l,mid);
    add(now<<1|1,mid+1,r);
    up(now);
    return;
}
void change(int now,int x,int v)
{
    int mid=(tree[now].l+tree[now].r)>>1;
    if(tree[now].l==x && tree[now].r==x)
    {
        tree[now].sum=tree[now].maxx=v;
        return;
    }
    if(x<=mid) change(now<<1,x,v);
    else change(now<<1|1,x,v);
    up(now);
    return;
}
int ask_sum(int now,int l,int r)
{
    int mid=(tree[now].l+tree[now].r)>>1,ans=0;
    if(tree[now].l>=l && tree[now].r<=r) return tree[now].sum;
    if(l<=mid) ans+=ask_sum(now<<1,l,r);
    if(r>=mid+1) ans+=ask_sum(now<<1|1,l,r);
    return ans;
}
int ask_max(int now,int l,int r)
{
    int mid=(tree[now].l+tree[now].r)>>1,ans=-inf;
    if(tree[now].l>=l && tree[now].r<=r) return tree[now].maxx;
    if(l<=mid) ans=max(ans,ask_max(now<<1,l,r));
    if(r>=mid+1) ans=max(ans,ask_max(now<<1|1,l,r));
    return ans;
}
int find_sum(int x,int y)
{
    int fx=top[x],fy=top[y],ans=0;
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        ans+=ask_sum(1,inseg[fx],inseg[x]);
        x=fa[fx],fx=top[x];
    }
    if(deep[x]>deep[y]) swap(x,y);
    ans+=ask_sum(1,inseg[x],inseg[y]);
    return ans;
}
int find_max(int x,int y)
{
    int fx=top[x],fy=top[y],ans=-inf;
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        ans=max(ans,ask_max(1,inseg[fx],inseg[x]));
        x=fa[fx],fx=top[x];
    }
    if(deep[x]>deep[y]) swap(x,y);
    ans=max(ans,ask_max(1,inseg[x],inseg[y]));
    return ans;
}
void solve()
{
    cin>>n;
    int x,y,q;
    string s;
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),build(x,y),build(y,x);
    for(int i=1;i<=n;i++) scanf("%d",&num[i]);
    dfs1(1,0),dfs2(1,1),add(1,1,n);
    cin>>q;
    while(q--)
    {
        cin>>s;
        scanf("%d%d",&x,&y);
        if(s=="QMAX") printf("%d\n",find_max(x,y));
        else if(s=="QSUM") printf("%d\n",find_sum(x,y));
        else change(1,inseg[x],y),num[x]=y;
    }
    return;
}
int main()
{
    solve();
    return 0;
}