[BZOJ 4034] 树上操作

Link:

BZOJ 4034 传送门

Solution：

树剖模板题……

Code:

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
#define mid ((l+r)>>1)
#define lc k<<1,l,mid
#define rc k<<1|1,mid+1,r
const int MAXN=1e5+;
struct edge{int to,nxt;}e[MAXN<<];
ll seg[MAXN<<],tag[MAXN<<];
int n,m,x,y,head[MAXN],dep[MAXN],dat[MAXN],sz[MAXN],top[MAXN],pos[MAXN],f[MAXN],cnt,tot;

void add_edge(int from,int to)
{e[++tot].nxt=head[from];e[tot].to=to;head[from]=tot;}

void dfs1(int x)
{
    sz[x]=;
    for(int i=head[x];i;i=e[i].nxt)
    {
        if(e[i].to==f[x]) continue;
        f[e[i].to]=x;dep[e[i].to]=dep[x]+;
        dfs1(e[i].to);sz[x]+=sz[e[i].to];
    }
}

void dfs2(int x,int up)
{
    int bs=;top[x]=up;pos[x]=++cnt;
    for(int i=head[x];i;i=e[i].nxt)
        if(e[i].to!=f[x]&&sz[e[i].to]>sz[bs]) bs=e[i].to;
    if(!bs) return;
    dfs2(bs,up);
    for(int i=head[x];i;i=e[i].nxt)
        if(e[i].to!=f[x]&&e[i].to!=bs) dfs2(e[i].to,e[i].to);
}

void pushdown(int k,int l,int r)
{
    if(!tag[k]) return;
    tag[k<<]+=tag[k];tag[k<<|]+=tag[k];
    seg[k<<]+=tag[k]*(mid-l+);
    seg[k<<|]+=tag[k]*(r-mid);
    tag[k]=;
}

void Update(int a,int b,int x,int k,int l,int r)
{
    if(a<=l&&r<=b)
    {seg[k]+=1ll*x*(r-l+);tag[k]+=x;return;}

    pushdown(k,l,r);
    if(a<=mid) Update(a,b,x,lc);
    if(b>mid) Update(a,b,x,rc);
    seg[k]=seg[k<<]+seg[k<<|];
}

ll Query(int a,int b,int k,int l,int r)
{
    if(a<=l&&r<=b) return seg[k];
    ll ret=;pushdown(k,l,r);
    if(a<=mid) ret+=Query(a,b,lc);
    if(b>mid) ret+=Query(a,b,rc);
    return ret;
}

ll solve_query(int a,int b)
{
    ll ret=;
    while(top[a]!=top[b])
    {
        if(dep[top[a]]<dep[top[b]]) swap(a,b);
        ret+=Query(pos[top[a]],pos[a],,,n);
        a=f[top[a]];
    }
    if(pos[a]>pos[b]) swap(a,b);
    ret+=Query(pos[a],pos[b],,,n);
    return ret;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%d",&dat[i]);
    for(int i=;i<n;i++)
        scanf("%d%d",&x,&y),add_edge(x,y),add_edge(y,x);
    dfs1();dfs2(,);
    for(int i=;i<=n;i++) Update(pos[i],pos[i],dat[i],,,n);

    while(m--)
    {
        int op,x,a;scanf("%d%d",&op,&x);
        if(op!=) scanf("%d",&a);
        if(op==) Update(pos[x],pos[x],a,,,n);
        else if(op==) Update(pos[x],pos[x]+sz[x]-,a,,,n);
        else printf("%lld\n",solve_query(,x));
    }
    return ;
}