【bzoj1036】[ZJOI2008]树的统计Count

题意：

I. CHANGE u t : 把结点u的权值改为t

II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值

III. QSUM u v: 询问从点u到点v的路径上的节点的权值和

树链剖分

siz[v]表示以v为根的子树的节点数

top[v]表示v所在的重链的顶端节点

fa[v]表示v的父亲

son[v]表示与v在同一重链上的v的儿子节点

pos[v]表示v与其父亲节点的连边在线段树中的位置

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
 
typedef long long LL;
 
#define INF 0x7fffffff
#define N 30010
 
struct Node
{
    LL maxn,sum;
}tr[N<<2];
 
struct edge
{
    int to,next;
}e[N<<2];
 
int cnt,head[N<<2];
 
int id,siz[N],top[N],fa[N],son[N];
int d[N],pos[N],v[N];
 
char s[15];
 
int n,m;
int a,b;
 
LL ans,sum;
 
void link(int x,int y)
{
    e[++cnt]=(edge){y,head[x]};
    head[x]=cnt;
}
 
void dfs(int x,int y)
{
    siz[x]=1;
    fa[x]=y;
    d[x]=d[y]+1;
    int k=0;
    for (int i=head[x];i;i=e[i].next)
        if (e[i].to!=fa[x])
        {
            dfs(e[i].to,x);
            siz[x]+=siz[e[i].to];
            if (siz[e[i].to]>k)
                k=siz[e[i].to],son[x]=e[i].to;
        }
}
 
void dfs2(int x,int y)
{
    top[x]=y;
    pos[x]=++id;
    if (!son[x])
        return ;
    dfs2(son[x],y);
    for (int i=head[x];i;i=e[i].next)
        if (e[i].to!=fa[x] && e[i].to!=son[x])
            dfs2(e[i].to,e[i].to);
}
 
void pushup(int now)
{
    tr[now].sum=tr[now<<1].sum+tr[now<<1|1].sum;
    tr[now].maxn=max(tr[now<<1].maxn,tr[now<<1|1].maxn);
}
 
void update(int nowl,int nowr,int now,int x,int y)
{
    if (nowl==nowr)
    {
        tr[now].sum=y;
        tr[now].maxn=y;
        return ;
    }
    int mid=(nowl+nowr)>>1;
    if (x<=mid)
        update(nowl,mid,now<<1,x,y);
    else
        update(mid+1,nowr,now<<1|1,x,y);
    pushup(now);
}
 
void query(int nowl,int nowr,int now,int x,int y)
{
    if (nowl==x && nowr==y)
    {
        ans=max(ans,tr[now].maxn);
        sum+=tr[now].sum;
        return ; 
    }
    int mid=(nowl+nowr)>>1;
    if (y<=mid)
        query(nowl,mid,now<<1,x,y);
    else if (x>mid)
        query(mid+1,nowr,now<<1|1,x,y);
    else
        query(nowl,mid,now<<1,x,mid),query(mid+1,nowr,now<<1|1,mid+1,y);
}
 
void find(int x,int y)
{
    int f1=top[x],f2=top[y];
    ans=-INF;
    sum=0;
    while (f1!=f2)
    {
        if (d[f1]<d[f2])
            swap(f1,f2),swap(x,y);
        query(1,n,1,pos[f1],pos[x]);
        x=fa[f1];
        f1=top[x];
    }
    if (d[x]>d[y])
        swap(x,y);
    query(1,n,1,pos[x],pos[y]);
}
 
int main()
{
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        link(a,b);
        link(b,a);
    }
    for (int i=1;i<=n;i++)
        scanf("%d",&v[i]);
    d[1]=1;
    dfs(1,0);
    dfs2(1,1);
    for (int i=1;i<=n;i++)
        update(1,n,1,pos[i],v[i]);
    scanf("%d",&m);
    int x,y;
    while (m--)
    {
        scanf("%s%d%d",s,&x,&y);
        if (s[0]=='C')
            update(1,n,1,pos[x],y);
        else
        {
            find(x,y);
            if (s[1]=='M')
                printf("%lld
",ans);
            else
                printf("%lld
",sum);
        }
    }
    return 0;
}