BZOJ 4034 [HAOI2015]树上操作 - 树链剖分

大概是一道裸题。注意处理给所有子树add时可以采用给入时间和出时间的区域线段树统一加add的方法，因为子树包含且仅包含在这一段区域内。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
 
using namespace std;
 
const int maxn=100005;
 
struct edge
{
    int to,next;
}e[maxn<<1];
struct tree
{
    long long val,add;
}t[maxn<<2];
 
int n,q,dfs_clock,cnt;
int s[maxn],f[maxn],id[maxn],out_time[maxn],head[maxn];
int size[maxn],son[maxn],per[maxn],top[maxn];
 
inline void insert(int a,int b)
{
    e[++cnt].to=b;e[cnt].next=head[a];head[a]=cnt;
}
inline void maintain(int ro)
{
    t[ro].val=t[ro<<1].val+t[ro<<1|1].val;
}
void build(int ro,int l,int r)
{
    if(l==r)
    {
        t[ro].val=s[f[l]];
        return;
    }
    int mid=l+r>>1;
    build(ro<<1,l,mid);
    build(ro<<1|1,mid+1,r);
    maintain(ro);
}
inline void pushdown(int ro,int l,int r)
{
    int mid=l+r>>1;
    if(t[ro].add)
    {
        t[ro<<1].add+=t[ro].add;
        t[ro<<1|1].add+=t[ro].add;
        t[ro<<1].val+=1LL*(mid-l+1)*t[ro].add;
        t[ro<<1|1].val+=1LL*(r-mid)*t[ro].add;
        t[ro].add=0;
    }
}
void addition(int ro,int L,int R,int l,int r,int add)
{
    if(L==l&&R==r)
    {
        t[ro].val+=1LL*(r-l+1)*add;
        t[ro].add+=add;
        return;
    }
    pushdown(ro,l,r);
    int mid=l+r>>1;
    if(R<=mid)addition(ro<<1,L,R,l,mid,add);
    else if(L>=mid+1)addition(ro<<1|1,L,R,mid+1,r,add);
    else addition(ro<<1,L,mid,l,mid,add),addition(ro<<1|1,mid+1,R,mid+1,r,add);
    maintain(ro);
}
long long query(int ro,int L,int R,int l,int r)
{
    if(L==l&&R==r)
        return t[ro].val;
    pushdown(ro,l,r);
    int mid=l+r>>1;
    if(R<=mid)return query(ro<<1,L,R,l,mid);
    else if(L>=mid+1)return query(ro<<1|1,L,R,mid+1,r);
    else return query(ro<<1,L,mid,l,mid)+query(ro<<1|1,mid+1,R,mid+1,r);
}
void addition(int x,int add)
{
    addition(1,id[x],out_time[x],1,n,add);
}
long long query(int x)
{
    long long res=0;
    int tp=top[x];
    while(x)
    {
        res+=query(1,id[tp],id[x],1,n);
        x=per[tp];
        tp=top[x];
    }
    return res;
}
void dfs(int x,int fa)
{
    size[x]=1;
    per[x]=fa;
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=fa)
    {
        dfs(e[i].to,x);
        size[x]+=size[e[i].to];
        if(size[e[i].to]>size[son[x]])
            son[x]=e[i].to;
    }
}
void dfs_chain(int x,int tp)
{
    id[x]=++dfs_clock;//节点映射编号 
    f[dfs_clock]=x;//编号映射节点 
    top[x]=tp;
    if(son[x])
        dfs_chain(son[x],tp);
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=per[x]&&e[i].to!=son[x])
            dfs_chain(e[i].to,e[i].to);
    out_time[x]=dfs_clock;
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
        scanf("%d",s+i);
    for(int i=1;i<n;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        insert(a,b);
        insert(b,a);
    }
    dfs(1,0);
    dfs_chain(1,1);
    build(1,1,n);
    while(q--)
    {
        int d,d1,d2;
        scanf("%d",&d);
        if(d==1)
        {
            scanf("%d%d",&d1,&d2);
            addition(1,id[d1],id[d1],1,n,d2);
        }
        else if(d==2)
        {
            scanf("%d%d",&d1,&d2);
            addition(d1,d2);
        }
        else if(d==3)
        {
            scanf("%d",&d1);
            printf("%lld\n",query(d1));
        }
    }
    return 0;
}