树链剖分（复习+代码解释）

• 将树上的链通过dfs序映射成一个区间，显然一个子树对应一个连续区间，一条重链对应一个连续区间，一条链可以被分成不超过logn个区间，通过用数据结构（如线段树）维护这些区间可以实现nlogn的维护这些树链

• 树上操作

#include <iostream>
#include <algorithm>
#include <fstream>
using namespace std;
const int maxn=1e5+10;
int n,m,a[maxn],last[maxn],cnt,xx,yy,son[maxn],dep[maxn],fa[maxn],sizes[maxn],dfn[maxn],tot;//son:重儿子，sizes:子树大小
long long tops[maxn],b[maxn],lk[maxn],tree[4*maxn],tag[4*maxn],mak,ans;//tops:重链的根 a:线段树处理的点(线段树处理的区间)，与树上的点映射 tree:线段树上的点
struct edge{
    int v,next;
}e[maxn*2];
inline void add(int u,int v)
{
    e[++cnt].v=v;
    e[cnt].next=last[u];
    last[u]=cnt;
}
void dfs1(int u,int f)
{
    sizes[u]=1;
    dep[u]=dep[f]+1;
    fa[u]=f;
    for(int i=last[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==f)
        continue;
        dfs1(v,u);
        sizes[u]+=sizes[v];
        if(sizes[v]>sizes[son[u]])
        son[u]=v;
    }
}
void dfs2(int u,int tp)//tp:u所在的重链的根
{
    tops[u]=tp;
    dfn[u]=++tot;
    a[tot]=u;
    if(son[u])
    dfs2(son[u],tp);
    for(int i=last[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v!=son[u]&&v!=fa[u])
        dfs2(v,v);
    }
    lk[u]=tot;//lk:该子树映射所有结点的最大dfs序，这样就知道了每颗子树在a上的映射区间了
}
void pushup(int x)
{
    tree[x]=tree[x<<1]+tree[x<<1|1];
}
void pushdown(int x,int l,int r)//tag:为其儿子打标记
{
    int ll=x<<1,rr=x<<1|1,mid=(l+r)>>1;//ll线段树左儿子,rr线段树右儿子
    tree[ll]+=(mid-l+1)*tag[x];
    tree[rr]+=(r-mid)*tag[x];
    tag[ll]+=tag[x];
    tag[rr]+=tag[x];
    tag[x]=0;
}
void build(int x,int l,int r)
{
    if(l==r)
    {
        tree[x]=b[a[l]];
        return;
    }
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    pushup(x);
}
void update(int x,int l,int r,int kl,int kr,long long k)
{
    if(l>kr||r<kl) return;
    if(l>=kl&&r<=kr)
    {
        tree[x]+=(r-l+1)*k;
        tag[x]+=k;
        return;
    }
    int mid=(l+r)>>1;
    pushdown(x,l,r);
    update(x<<1,l,mid,kl,kr,k);
    update(x<<1|1,mid+1,r,kl,kr,k);
    pushup(x);
}
void query(int x,int l,int r,int kl,int kr)
{
    if(l>kr||r<kl) return;
    if(l>=kl&&r<=kr)
    {
        ans+=tree[x];
        return ;
    }
    pushdown(x,l,r);
    int mid=(l+r)>>1;
    query(x<<1,l,mid,kl,kr);
    query(x<<1|1,mid+1,r,kl,kr);
}
void cal(int x,int y,int z,long long w)//x,y点，z操作 w权值
{
    while(tops[x]!=tops[y])
    {
        if(dep[tops[x]]<dep[tops[y]]) swap(x,y);
        if(z==2)
        update(1,1,n,dfn[tops[x]],dfn[x],w);
        else
        query(1,1,n,dfn[tops[x]],dfn[x]);
        x=fa[tops[x]];
    }
    if(dep[x]<dep[y]) swap(x,y);
    if(z==2)
    update(1,1,n,dfn[y],dfn[x],w);
    else
    query(1,1,n,dfn[y],dfn[x]);
}
int main()
{
//    freopen("1.txt","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%lld",&b[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&xx,&yy);
        add(xx,yy);
        add(yy,xx);
    }
    dfs1(1,0);dfs2(1,1);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&mak);
        if(mak==1)
        {
            scanf("%d%d",&xx,&yy);
            update(1,1,n,dfn[xx],dfn[xx],yy);
        }
        if(mak==2)
        {
            scanf("%d%d",&xx,&yy);
            update(1,1,n,dfn[xx],lk[xx],yy);
        }
        if(mak==3)
        {
            scanf("%d",&xx);
            cal(xx,1,3,0);
            cout<<ans<<endl;
            ans=0;
        }
    }
    return 0;
}