bzoj 1036: [ZJOI2008]树的统计Count

树剖板子

然而不停WA....

注意：

1.分清pos[a],arr[a];曾经116行pos[a]写成a，WA

2.134行曾经复制的时候没改，成了qmax(..)

3.126,137行这里没有a!=b或其他的判断语句，直接查询

4.133行不是dep[a]<dep[b]，这里是比较链顶点的深度，相当于比较两个点向上跳一次之后的深度，而不是自身的深度

5.46行不要写成maxn[l]，a[num],a[l],...注意线段树上点的下标是num，原数据一定要按a[arr[l]]取

6.各个取最大值的函数初始值要为-inf，而各个取和的函数初始值要为0

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long LL;
struct E
{
    LL to,nxt;
}e[60100];
LL f1[30100],ne;
LL hson[30100],f[30100],sz[30100],arr[30100],pos[30100],top[30100],dep[30100];
LL aa[30100];
void dfs1(LL u,LL fa)
{
    sz[u]=1;
    for(LL k=f1[u];k;k=e[k].nxt)
        if(e[k].to!=fa)
        {
            dep[e[k].to]=dep[u]+1;
            dfs1(e[k].to,u);
            sz[u]+=sz[e[k].to];
            if(sz[hson[u]]<sz[e[k].to])    hson[u]=e[k].to;
        }
}
void dfs2(LL u,LL fa)
{
    arr[++arr[0]]=u;pos[u]=arr[0];
    f[u]=fa;
    if(u==hson[fa])    top[u]=top[fa];
    else    top[u]=u;
    if(hson[u])    dfs2(hson[u],u);
    for(LL k=f1[u];k;k=e[k].nxt)
        if(e[k].to!=fa&&e[k].to!=hson[u])
        {
            dfs2(e[k].to,u);
        }
}
LL n,q;
namespace SegT
{
#define mid (l+((r-l)>>1))
#define lc (num<<1)
#define rc (num<<1|1)
    LL maxn[120100],sum[120100];
    void build(LL l,LL r,LL num)
    {
        if(l==r)    {maxn[num]=sum[num]=aa[arr[l]];return;}
        build(l,mid,lc);build(mid+1,r,rc);
        maxn[num]=max(maxn[lc],maxn[rc]);
        sum[num]=sum[lc]+sum[rc];
    }
    LL L,R,x;
    void _update(LL l,LL r,LL num)
    {
        if(l==r)    {maxn[num]=sum[num]=x;return;}
        if(L<=mid)    _update(l,mid,lc);
        else    _update(mid+1,r,rc);
        maxn[num]=max(maxn[lc],maxn[rc]);
        sum[num]=sum[lc]+sum[rc];
    }
    LL _qmax(LL l,LL r,LL num)
    {
        if(L<=l&&r<=R)    return maxn[num];
        LL ans=-0x3f3f3f3f;
        if(L<=mid)    ans=max(ans,_qmax(l,mid,lc));
        if(mid<R)    ans=max(ans,_qmax(mid+1,r,rc));
        return ans;
    }
    LL _qsum(LL l,LL r,LL num)
    {
        if(L<=l&&r<=R)    return sum[num];
        LL ans=0;
        if(L<=mid)    ans+=_qsum(l,mid,lc);
        if(mid<R)    ans+=_qsum(mid+1,r,rc);
        return ans;
    }
    void update(LL a,LL b)
    {
        L=a;x=b;_update(1,n,1);
    }
    LL qmax(LL l,LL r)
    {
        L=l;R=r;if(L>R)    swap(L,R);
        return _qmax(1,n,1);
    }
    LL qsum(LL l,LL r)
    {
        L=l;R=r;if(L>R)    swap(L,R);
        return _qsum(1,n,1);
    }
#undef mid
#undef lc
#undef rc
}
using SegT::qmax;
using SegT::qsum;
int main()
{
    LL i,a,b,ans;char tmp[10];
    scanf("%lld",&n);
    for(i=1;i<n;i++)
    {
        scanf("%lld%lld",&a,&b);
        e[++ne].to=b;e[ne].nxt=f1[a];f1[a]=ne;
        e[++ne].to=a;e[ne].nxt=f1[b];f1[b]=ne;
    }
    for(i=1;i<=n;i++)    scanf("%lld",&aa[i]);
    dfs1(1,0);dfs2(1,0);
    SegT::build(1,n,1);
    scanf("%lld",&q);
    while(q--)
    {
        scanf("%s%lld%lld",tmp,&a,&b);
        switch(tmp[1])
        {
        case 'H':
            SegT::update(pos[a],b);
            break;
        case 'M':
            ans=-0x3f3f3f3f;
            while(top[a]!=top[b])
            {
                if(dep[top[a]]<dep[top[b]])    swap(a,b);
                ans=max(ans,qmax(pos[a],pos[top[a]]));
                a=f[top[a]];
            }
            ans=max(ans,qmax(pos[a],pos[b]));
            printf("%lld
",ans);
            break;
        case 'S':
            ans=0;
            while(top[a]!=top[b])
            {
                if(dep[top[a]]<dep[top[b]])    swap(a,b);//不是dep[a]<dep[b]
                ans+=qsum(pos[a],pos[top[a]]);//曾经不小心复制错，成了qmax(...)
                a=f[top[a]];
            }
            ans+=qsum(pos[a],pos[b]);//没有if(a!=b)
            printf("%lld
",ans);
            break;
        }
    }
    return 0;
}