[ZJOI2008] 树的统计

洛谷 P2590 传送门

bzoj 1036 传送门

感觉最近有点虚，码个板子爽一下。

居然没一A，交了两遍才A......

其原因是定义qmx函数的时候我懒得重敲一遍，直接把qsum那个复制过来改了一下。

然后忘把qsum(p<<1,l,r)和qsum(p<<1|1,l,r)改成qmx了。

这样的话只要是qmx操作，第一遍还是qmx，分治到两边就变成qsum了......

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int n,q;
int hd[30005],to[60005],nx[60005],ec;
int bv[30005];

void edge(int af,int at)
{
    to[++ec]=at;
    nx[ec]=hd[af];
    hd[af]=ec;
}

int f[30005],dep[30005],tp[30005];
int sz[30005],son[30005];
int in[30005],out[30005],pc,pos[30005];

void pre(int p,int fa)
{
    f[p]=fa;
    dep[p]=dep[fa]+1;
    sz[p]=1;
    for(int i=hd[p];i;i=nx[i])
    {
        if(to[i]==fa)continue;
        pre(to[i],p);
        sz[p]+=sz[to[i]];
        if(sz[son[p]]<sz[to[i]])son[p]=to[i];
    }
}

void dfs(int p)
{
    in[p]=++pc;
    pos[pc]=p;
    if(p==son[f[p]])tp[p]=tp[f[p]];
    else tp[p]=p;
    if(son[p])dfs(son[p]);
    for(int i=hd[p];i;i=nx[i])
        if(to[i]!=f[p]&&to[i]!=son[p])dfs(to[i]);
    out[p]=pc;
}

int sum[120005],mx[120005],lb[120005],rb[120005];

void pushup(int p)
{
    sum[p]=sum[p<<1]+sum[p<<1|1];
    mx[p]=max(mx[p<<1],mx[p<<1|1]);
}

void build(int p,int l,int r)
{
    lb[p]=l,rb[p]=r;
    if(l==r)
    {
        mx[p]=sum[p]=bv[pos[l]];
        return;
    }
    int mid=(l+r)>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    pushup(p);
}

void change(int p,int pnt,int v)
{
    if(lb[p]==rb[p])
    {
        sum[p]=mx[p]=v;
        return;
    }
    int mid=(lb[p]+rb[p])>>1;
    if(pnt<=mid)change(p<<1,pnt,v);
    else change(p<<1|1,pnt,v);
    pushup(p);
}

int qsum(int p,int l,int r)
{
    if(lb[p]>=l&&rb[p]<=r)return sum[p];
    int ret=0,mid=(lb[p]+rb[p])>>1;
    if(l<=mid)ret+=qsum(p<<1,l,r);
    if(r>mid)ret+=qsum(p<<1|1,l,r);
    return ret;
}

int qmx(int p,int l,int r)
{
    if(lb[p]>=l&&rb[p]<=r)return mx[p];
    int ret=-0x3f3f3f3f,mid=(lb[p]+rb[p])>>1;
    if(l<=mid)ret=max(qmx(p<<1,l,r),ret);
    if(r>mid)ret=max(qmx(p<<1|1,l,r),ret);
    return ret;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int ff,tt;
        scanf("%d%d",&ff,&tt);
        edge(ff,tt);
        edge(tt,ff);
    }
    for(int i=1;i<=n;i++)scanf("%d",&bv[i]);
    pre(1,1);
    dfs(1);
    build(1,1,n);
    scanf("%d",&q);
    while(q--)
    {
        char op[15];
        scanf("%s",op+1);
        if(op[2]=='H')
        {
            int u,t;
            scanf("%d%d",&u,&t);
            change(1,in[u],t);
        }
        if(op[2]=='M')
        {
            int x,y;
            scanf("%d%d",&x,&y);
            int ans=-0x3f3f3f3f;
            while(tp[x]!=tp[y])
            {
                if(dep[tp[x]]<dep[tp[y]])swap(x,y);
                ans=max(qmx(1,in[tp[x]],in[x]),ans);
                x=f[tp[x]];
            }
            if(dep[x]>dep[y])swap(x,y);
            ans=max(qmx(1,in[x],in[y]),ans);
            printf("%d\n",ans);
        }
        if(op[2]=='S')
        {
            int x,y;
            scanf("%d%d",&x,&y);
            int ans=0;
            while(tp[x]!=tp[y])
            {
                if(dep[tp[x]]<dep[tp[y]])swap(x,y);
                ans+=qsum(1,in[tp[x]],in[x]);
                x=f[tp[x]];
            }
            if(dep[x]>dep[y])swap(x,y);
            ans+=qsum(1,in[x],in[y]);
            printf("%d\n",ans);
        }
    }
    return 0;
}