bzoj4034 [HAOI2015]树上操作——树链剖分

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4034

树剖裸题；

一定要注意 long long ！！！

update 的时候别忘了 pushdown ...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int const maxn=1e5+;
int n,m,tim,dfn[maxn],id[maxn],top[maxn],son[maxn],siz[maxn],fa[maxn];
int hd[maxn],ct,a[maxn],end[maxn];
ll sum[maxn<<],lzy[maxn<<];//ll
struct N{
    int to,nxt;
    N(int t=,int n=):to(t),nxt(n) {}
}ed[maxn<<];
void add(int x,int y){ed[++ct]=N(y,hd[x]); hd[x]=ct;}
void dfs(int x)
{
    siz[x]=;
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa[x])continue;
        fa[u]=x; dfs(u); siz[x]+=siz[u];
        if(siz[u]>siz[son[x]])son[x]=u;
    }
}
void dfs2(int x)
{
    id[dfn[x]=++tim]=x;
    if(son[x])top[son[x]]=top[x],dfs2(son[x]);
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa[x]||u==son[x])continue;
        top[u]=u; dfs2(u);
    }
    end[x]=tim;
}
void pushup(int x){sum[x]=sum[x<<]+sum[x<<|];}
void pushdown(int x,int l,int r)
{
    if(lzy[x])
    {
        int ls=(x<<),rs=(x<<|),mid=((l+r)>>);
        sum[ls]+=lzy[x]*(mid-l+); lzy[ls]+=lzy[x];
        sum[rs]+=lzy[x]*(r-mid); lzy[rs]+=lzy[x];
        lzy[x]=;
    }
}
void build(int x,int l,int r)
{
    if(l==r){sum[x]=a[id[l]]; return;}
    int mid=((l+r)>>);
    build(x<<,l,mid); build(x<<|,mid+,r);
    pushup(x);
}
ll query(int x,int l,int r,int L,int R)
{
    if(l>=L&&r<=R)return sum[x];
    pushdown(x,l,r);
    int mid=((l+r)>>); ll ret=;
    if(mid>=L)ret+=query(x<<,l,mid,L,R);
    if(mid<R)ret+=query(x<<|,mid+,r,L,R);
    return ret;
}
void update(int x,int l,int r,int L,int R,ll v)//ll
{
    if(l>=L&&r<=R){sum[x]+=v*(r-l+); lzy[x]+=v; return;}
    pushdown(x,l,r);//!!!
    int mid=((l+r)>>);
    if(mid>=L)update(x<<,l,mid,L,R,v);
    if(mid<R)update(x<<|,mid+,r,L,R,v);
    pushup(x);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++)scanf("%d",&a[i]);
    for(int i=,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    dfs(); top[]=; dfs2(); build(,,n);
    for(int i=,p,x;i<=m;i++)
    {
        ll v;//
        scanf("%d%d",&p,&x);
        if(p==){scanf("%lld",&v); update(,,n,dfn[x],dfn[x],v);}
        if(p==){scanf("%lld",&v); update(,,n,dfn[x],end[x],v);}
        if(p==)
        {
            ll ans=;
            while(x)//
            {
                ans+=query(,,n,dfn[top[x]],dfn[x]);//顺序
                x=fa[top[x]];
            }
            printf("%lld\n",ans);
        }
    }
    return ;
}