【模板】 树链剖分

#include<bits/stdc++.h>
#define N 100010
#define INF 0x3f3f3f3f
#define eps 1e-10
#define pi 3.141592653589793
#define P 1000000007 
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define mem(x) memset(x,0,sizeof x)
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
vector<int> a[N];
int n,m,cnt,rt,tg,son[N],top[N],id[N],fa[N],d[N],sz[N],rk[N],w[N],mod;
struct node{ int s,lz; }f[N<<2];
void dfs1(int x,int ffa){
    sz[x]=1;
    for(auto i:a[x]) if (i!=ffa){
        fa[i]=x; d[i]=d[x]+1; dfs1(i,x); sz[x]+=sz[i];
        if (sz[i]>sz[son[x]]) son[x]=i;
    }
}
void dfs2(int x,int t){
    top[x]=t; id[x]=++cnt; rk[cnt]=x;
    if (!son[x]) return;
    dfs2(son[x],t);
    for (auto i:a[x]) if (i!=son[x] && i!=fa[x]) dfs2(i,i);
}

inline void pushdown(int x,int l,int r,int t){
    if (f[x].lz){
        f[x<<1].lz+=f[x].lz; f[x<<1].s+=(t-l+1)*f[x].lz;
        f[x<<1|1].lz+=f[x].lz; f[x<<1|1].s+=(r-t)*f[x].lz;
        f[x].lz=0;
    }
}

void updata(int x,int l,int r,int fl,int fr){
    if (l==fl && r==fr){
        f[x].lz+=tg; f[x].s+=(r-l+1)*tg;
    }else{
        int t=l+r>>1;
        pushdown(x,l,r,t);
        if (fr<=t) updata(x<<1,l,t,fl,fr);else
        if (fl>t) updata(x<<1|1,t+1,r,fl,fr);else{
            updata(x<<1,l,t,fl,t);
            updata(x<<1|1,t+1,r,t+1,fr);
        }
        f[x].s=f[x<<1].s+f[x<<1|1].s;
    }
}

int query(int x,int l,int r,int fl,int fr){
    if (l==fl && r==fr) return f[x].s;
    int t=l+r>>1;
    pushdown(x,l,r,t);
    if (fr<=t)return query(x<<1,l,t,fl,fr);else
    if (fl>t)return query(x<<1|1,t+1,r,fl,fr);else
    return query(x<<1,l,t,fl,t)+query(x<<1|1,t+1,r,t+1,fr);
}

int sum(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if (d[top[x]]<d[top[y]]) swap(x,y);
        ans+=query(1,1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if (d[x]>d[y]) swap(x,y);
    ans+=query(1,1,n,id[x],id[y]);
    return ans;
}
void change(int x,int y){
    while(top[x]!=top[y]){
        if (d[top[x]]<d[top[y]]) swap(x,y);
        updata(1,1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if (d[x]>d[y]) swap(x,y);
    updata(1,1,n,id[x],id[y]);
}
int main(){
    scc(n,m); scc(rt,mod);
    for (int i=1;i<=n;i++) sc(w[i]);
    for (int i=1,x,y;i<n;i++){
        scc(x,y);
        a[x].pb(y),a[y].pb(x);
    }
    dfs1(rt,-1);
    dfs2(rt,rt);
    for (int i=1;i<=n;i++)
        tg=w[i],change(i,i);          //初始值
    for (int i=1,op,x,y;i<=m;i++){
        sc(op);
        if (op==1){
            sccc(x,y,tg);             //x->y路径上加tg
            change(x,y);
        }else if (op==2){
            scc(x,y);                  //询问x->y路径上的和
            printf("%d\n",sum(x,y));
        }else if (op==3){
            scc(x,tg);                  //x为根的子树加tg
            updata(1,1,n,id[x],id[x]+sz[x]-1);
        }else{
            sc(x);                       //询问x为根子树上的和
            printf("%d\n",query(1,1,n,id[x],id[x]+sz[x]-1));
        }
    }
}