LOJ10138

ZJOI 2008 树上的统计

一树上有 n 个节点，编号分别为 1 到 n，每个节点都有一个权值 w。我们将以下面的形式来要求你对这棵树完成一些操作：

1. CHANGE u t ：把节点 u 权值改为t；
2. QMAX u v ：询问点 u 到点 v 路径上的节点的最大权值；
3. QSUM u v ：询问点 u 到点 v 路径上的节点的权值和。

注意：从点 u 到点 v 路径上的节点包括 u 和 v 本身。

输入格式

第一行为一个数 n，表示节点个数；

接下来 n-1 行，每行两个整数 a,b，表示节点 a与节点 b 之间有一条边相连；

接下来 n 行，每行一个整数，第 i 行的整数 wi​ 表示节点i 的权值；

接下来一行，为一个整数 q ，表示操作总数；

接下来 q 行，每行一个操作，以 CHANGE u t 或 QMAX u v 或 QSUM u v的形式给出。

输出格式

对于每个 QMAX 或 QSUM 的操作，每行输出一个整数表示要求的结果。

样例

样例输入

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

样例输出

4
1
2
2
10
6
5
6
5
16

数据范围与提示

对于 100% 的数据，有1≤n≤3×10^4,0≤q≤2×10^5。中途操作中保证每个节点的权值 w 在 −30000 至 30000 之间。

_______________________________________________________________________________________

很简单的树剖，好久没有写了，第一次写的不知道哪里错了，还是写了第二次！

_______________________________________________________________________________________

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn=30020;
ll n,m;
struct edge
{
    ll u,v,nxt;
}e[maxn<<1];
ll head[maxn],js;
void addage(ll u,ll v)
{
    e[++js].u=u;e[js].v=v;
    e[js].nxt=head[u];head[u]=js;
}
ll w[maxn];
ll dep[maxn],fat[maxn],siz[maxn],son[maxn];
void dfs(ll u,ll fa)
{
    dep[u]=dep[fa]+1;
    fat[u]=fa;
    siz[u]=1;
    for(ll i=head[u];i;i=e[i].nxt)
    {
        ll v=e[i].v;
        if(v==fa)continue;
        dfs(v,u);
        siz[u]+=siz[v];
        if(!son[u] || siz[son[u]]<siz[v])son[u]=v;
    }
}
ll pos[maxn],top[maxn],fos[maxn],p;
void getpos(ll u,ll fa)
{
    top[u]=fa;
    pos[u]=++p;
    fos[p]=u;
    if(!son[u])return ;
    getpos(son[u],fa);
    for(ll i=head[u];i;i=e[i].nxt)
    {
        ll v=e[i].v;
        if(v!=son[u] && v!=fat[u])getpos(v,v);
    }
}
ll MAX,SUM;
ll _sum[maxn<<2],_max[maxn<<2];
void update(ll cur)
{
    _sum[cur]=_sum[cur<<1]+_sum[cur<<1|1];
    _max[cur]=max(_max[cur<<1],_max[cur<<1|1]);
}
void build(ll cur,ll l,ll r)
{
    if(l==r)
    {
        _sum[cur]=_max[cur]=w[fos[l]];
        return ;
    }
    ll mid=(l+r)>>1;
    build(cur<<1,l,mid);
    build(cur<<1|1,mid+1,r);
    update(cur);
}
void query(ll cur,ll l,ll r,ll ql,ll qr)
{
    if(ql<=l && r<=qr)
    {
        MAX=max(_max[cur],MAX);
        SUM+=_sum[cur];
        return ;
    }
    ll mid=(l+r)>>1;
    if(ql<=mid)query(cur<<1,l,mid,ql,qr);
    if(qr>mid)query(cur<<1|1,mid+1,r,ql,qr);
}
void ask(ll u,ll v)
{
    ll tpu=top[u],tpv=top[v];
    while(tpu!=tpv)
    {
        if(dep[tpu]<dep[tpv])
        {
            swap(u,v);swap(tpu,tpv);
        }
        query(1,1,n,pos[tpu],pos[u]);
        u=fat[tpu];tpu=top[u];
    }
    if(dep[u]>dep[v])swap(u,v);
    query(1,1,n,pos[u],pos[v]);
}
void change(ll cur,ll l,ll r,ll p,ll x)
{
    if(l==r)
    {
        _max[cur]=_sum[cur]=x;
        return ;
    }
    ll mid=(l+r)>>1;
    if(p<=mid)change(cur<<1,l,mid,p,x);
    else change(cur<<1|1,mid+1,r,p,x);
    update(cur);
}
int main()
{
    scanf("%lld",&n);
    for(ll i=1,u,v;i<n;++i)
    {
        scanf("%lld%lld",&u,&v);
        addage(u,v);addage(v,u);
    }
    for(int i=1;i<=n;++i)scanf("%lld",w+i);
    scanf("%lld",&m);
    dfs(1,0);
    getpos(1,1);
    build(1,1,n);
    char s[10];
    ll u,v;
    while(m--)
    {
        scanf("%s%lld%lld",s,&u,&v);
        if(s[1]=='H')change(1,1,n,pos[u],v);
        else
        {
            MAX=-0x7fffffff,SUM=0;
            ask(u,v);
            printf("%lld
",s[1]=='M'?MAX:SUM);
        }
    }
    return 0;
}