树链剖分模板

沃日，调了半天

代码:

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const ll maxn=1e5+9;
struct node{
    ll to,next;
}edge[maxn*2];
ll n,mod,cnt=0,value[maxn],id[maxn],rk[maxn],top[maxn],size[maxn],d[maxn],son[maxn],f[maxn],tot=0,head[maxn];
void add(ll u,ll v){
    edge[cnt].next=head[u];
    edge[cnt].to=v;
    head[u]=cnt++;
}
struct tree{
    ll l,r,lazy,sum;
}t[maxn*4];
void dfs1(ll u,ll pre){//预处理深度，父节点，树的大小，重儿子
    d[u]=d[pre]+1;
    size[u]=1;
    f[u]=pre;
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(v==pre)continue;
        dfs1(v,u);
        size[u]+=size[v];
        if(size[v]>size[son[u]])son[u]=v;
    }
}
void dfs2(ll u,ll t){//按dfs序给点设置编号，映射，处理重链起始点
    id[u]=++tot;
    rk[tot]=u;
    top[u]=t;
    if(son[u])
    dfs2(son[u],t);
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(v!=son[u]&&v!=f[u])dfs2(v,v);
    }
}
void pushup(ll rt){
    t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
    t[rt].sum%=mod;
}
void pushdown(ll rt){
    if(t[rt].lazy){
        t[rt<<1].lazy+=t[rt].lazy;
        t[rt<<1].lazy%=mod;
        t[rt<<1|1].lazy+=t[rt].lazy;
        t[rt<<1|1].lazy%=mod;
        t[rt<<1].sum+=t[rt].lazy*(t[rt<<1].r-t[rt<<1].l+1);
        t[rt<<1].sum%=mod;
        t[rt<<1|1].sum+=t[rt].lazy*(t[rt<<1|1].r-t[rt<<1|1].l+1);
        t[rt<<1|1].sum%=mod;
        t[rt].lazy=0;
    }
}
void build(ll l,ll r,ll rt){
    t[rt].l=l;t[rt].r=r;
    if(l==r){
        t[rt].sum=value[rk[l]]%mod;
        return;
    }
    ll mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(ll ul,ll ur,ll l,ll r,ll rt,ll c){
    if(l>=ul&&r<=ur){
        t[rt].sum=(t[rt].sum+(r-l+1)*c)%mod;
        t[rt].lazy=(t[rt].lazy+c)%mod;
        return;
    }//cout<<123<<endl;
    pushdown(rt);
    ll mid=(l+r)>>1;
    if(mid>=ul)update(ul,ur,l,mid,rt<<1,c);
    if(mid<ur)update(ul,ur,mid+1,r,rt<<1|1,c);
    pushup(rt);
}
void updates(ll x,ll y,ll c){
    while(top[x]!=top[y]){
        if(d[top[x]]<d[top[y]])swap(x,y);
        update(id[top[x]],id[x],1,n,1,c);
        x=f[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    update(id[x],id[y],1,n,1,c);
}
ll query(ll ql,ll qr,ll l,ll r,ll rt){
    if(l>=ql&&r<=qr){
        return t[rt].sum%mod;
    }

    pushdown(rt);
    ll mid=(l+r)>>1;
    ll sum=0;
    if(mid>=ql)sum+=query(ql,qr,l,mid,rt<<1);
    sum%=mod;
    if(mid<qr)sum+=query(ql,qr,mid+1,r,rt<<1|1);
    sum%=mod;
    return sum;
}
ll sum(ll x,ll y){
    ll ans=0;
    while(top[x]!=top[y]){
        if(d[top[x]]<d[top[y]])swap(x,y);
        ans+=query(id[top[x]],id[x],1,n,1);
        ans%=mod;
        x=f[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    ans+=query(id[x],id[y],1,n,1);
    ans%=mod;
    return ans;
}
int main(){
        memset(head,-1,sizeof(head));
        ll i,j,k,m,rt;
        scanf("%lld%lld%lld%lld",&n,&m,&rt,&mod);
        for(i=1;i<=n;i++){
            scanf("%lld",&value[i]);
        }
        ll op,x,y,z;
        for(i=0;i<n-1;i++){
            scanf("%lld%lld",&x,&y);
            add(x,y);add(y,x);
        }
        //init();
        dfs1(rt,rt);
        dfs2(rt,rt);
        build(1,n,1);
        for(i=0;i<m;i++){
            scanf("%lld",&op);
            if(op==1){
                scanf("%lld%lld%lld",&x,&y,&z);
                updates(x,y,z);
            }
            else if(op==2){
                scanf("%lld%lld",&x,&y);
                printf("%lld\n",sum(x,y));
            }
            else if(op==3){
                scanf("%lld%lld",&x,&z);
                update(id[x],id[x]+size[x]-1,1,n,1,z);
            }
            else if(op==4){
                scanf("%lld",&x);
                printf("%lld\n",query(id[x],id[x]+size[x]-1,1,n,1));
        }
    }
}