bzoj1036 树的统计 树链剖分

又是一道裸题，刚学会树链剖分，存个模版。

#include<iostream>
#include<cstdio>
using namespace std;
struct bian{
    int to;
}b[60005];
int fst[30005],nxt[60005];
int tot=1;
void build(int f,int t)
{
    b[++tot].to=t;
    nxt[tot]=fst[f];
    fst[f]=tot;
}
int sd[30005],fa[30005];
int sz[30005],zson[30005];
void dfs(int u)
{
    int v,ans=0;
    for(int i=fst[u];i;i=nxt[i])
    {
        v=b[i].to;
        if(!sd[v])
        {
            sd[v]=sd[u]+1;
            fa[v]=u;
            dfs(v);
            sz[u]+=sz[v];
            if(sz[v]>ans)
            {
                ans=sz[v];
                zson[u]=v;
            }
        }
    }
    sz[u]++;
}
int dfs_clock;
int ts[30005],top[30005];
int intr[30005];
void dfs_2(int u,int x)
{
    ts[u]=++dfs_clock;
    intr[dfs_clock]=u;
    top[u]=x;
    if(!zson[u])
        return ;
    dfs_2(zson[u],x);
    int v;
    for(int i=fst[u];i;i=nxt[i])
    {
        v=b[i].to;
        if(v==fa[u]||v==zson[u])
            continue;
        top[v]=v;
        dfs_2(v,v);
    }
}
int w[30005];
struct xds{
    int l,r,sum,maxx;
}tree[150005];
void up(int dq)
{
    tree[dq].sum=tree[dq<<1].sum+tree[dq<<1|1].sum;
    tree[dq].maxx=max(tree[dq<<1].maxx,tree[dq<<1|1].maxx);
}
void build(int dq,int l,int r)
{
    tree[dq].l=l;
    tree[dq].r=r;
    if(l==r)
    {
        tree[dq].sum=tree[dq].maxx=w[intr[l]];
        return ;
    }
    int mid=(l+r)>>1;
    build(dq<<1,l,mid);
    build(dq<<1|1,mid+1,r);
    up(dq);
}
void change(int dq,int l,int r,int d)
{
    if(tree[dq].l==l&&tree[dq].r==r)
    {
        tree[dq].sum=d;
        tree[dq].maxx=d;
        return ;
    }
    int mid=(tree[dq].l+tree[dq].r)>>1;
    if(mid>=l)  change(dq<<1,l,r,d);
    if(mid<r)   change(dq<<1|1,l,r,d);
    up(dq);
}
int inf=1e9;
int ask_max(int dq,int l,int r)
{
    if(tree[dq].l>=l&&tree[dq].r<=r)
        return tree[dq].maxx;
    int mid=(tree[dq].l+tree[dq].r)>>1;
    int ans=-inf;
    if(mid>=l)  ans=ask_max(dq<<1,l,r);
    if(mid<r)   ans=max(ans,ask_max(dq<<1|1,l,r));
    return ans;
}
int ask_sum(int dq,int l,int r)
{
    if(tree[dq].l>=l&&tree[dq].r<=r)
        return tree[dq].sum;
    int mid=(tree[dq].l+tree[dq].r)>>1;
    int ans=0;
    if(mid>=l)  ans+=ask_sum(dq<<1,l,r);
    if(mid<r)   ans+=ask_sum(dq<<1|1,l,r);
    return ans;
}
int find_max(int x,int y)
{
    int ans=-inf;
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(sd[fx]<sd[fy])
        {
            swap(x,y);
            swap(fx,fy);
        }
        ans=max(ans,ask_max(1,ts[fx],ts[x]));
        x=fa[fx];fx=top[x];
    }
    if(sd[x]>sd[y]) swap(x,y);
    ans=max(ans,ask_max(1,ts[x],ts[y]));
    return ans;
}
int find_sum(int x,int y)
{
    int ans=0;
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(sd[fx]<sd[fy])
        {
            swap(x,y);
            swap(fx,fy);
        }
        ans+=ask_sum(1,ts[fx],ts[x]);
        x=fa[fx];fx=top[x];
    }
    if(sd[x]>sd[y]) swap(x,y);
    ans+=ask_sum(1,ts[x],ts[y]);
    return ans;
}
char s[10];
int main()
{
    int n;
    scanf("%d",&n);
    int a,b;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        build(a,b);
        build(b,a);
    }
    sd[1]=1;fa[1]=1;
    dfs(1);top[1]=1;
    dfs_2(1,1);
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    build(1,1,n);
    int q;
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s%d%d",s,&a,&b);
        if(s[1]=='M')
            printf("%d\n",find_max(a,b));
        if(s[1]=='S')
            printf("%d\n",find_sum(a,b));
        if(s[1]=='H')
            change(1,ts[a],ts[a],b);
    }
}