树链剖分模板

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef long double dl;
const LL N =1e5+5;
int n,m,r,p;
int a[N],h[N],e[N*2],ne[N*2],num;
int dep[N],f[N],son[N],siz[N];
int idx[N],top[N],b[N],cnt;
void add(int x,int y)
{
    e[num]=y;ne[num]=h[x];h[x]=num++;
}
int dfs1(int u,int fa,int deep)
{
    dep[u]=deep;
    f[u]=fa;
    siz[u]=1;
    int maxson=-1;
    for(int i=h[u];~i;i=ne[i])
    {
        int v=e[i];
        if(v==fa)continue;
        siz[u]+=dfs1(v,u,deep+1);
        if(siz[v]>maxson) maxson=siz[v],son[u]=v;
    }
    return siz[u];
}
void dfs2(int u,int topfa)
{

    idx[u]=++cnt;
    a[cnt]=b[u];
    top[u]=topfa;
    if(!son[u]) return;
    dfs2(son[u],topfa);
    for(int i=h[u];~i;i=ne[i])
    {
        int v=e[i];
        if(!idx[v]) dfs2(v,v);
    }
}
struct segTree{
    int l,r,size,w,add;
}tr[N*4];
void pushup(int u)
{
    tr[u].w=((tr[u<<1].w+tr[u<<1|1].w)%p+p)%p;
}
void pushdown(int u)
{
    if(tr[u].add)
    {
        int x=tr[u].add;
        segTree &L=tr[u<<1],&R=tr[u<<1|1];
        L.w=((L.w+L.size*x%p+p)%p+p)%p;
        R.w=((R.w+R.size*x%p+p)%p+p)%p;
        L.add=((L.add+x)%p+p)%p;
        R.add=((R.add+x)%p+p)%p;
        tr[u].add=0;
    }
}
void build(int u,int l,int r)
{
    tr[u]={l,r,r-l+1};
    if(l==r) tr[u].w=(a[l]%p+p)%p;
    else
    {
        int mid=(l+r)>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
    }
    //cout<<l<<" "<<r<<" "<<tr[u].w<<endl;
}
void modify(int u,int l,int r,int x)
{
    //cout<<u<<endl;
    if(l<=tr[u].l&&r>=tr[u].r)
    {
        tr[u].add=((tr[u].add+x)%p+p)%p;
        tr[u].w=((tr[u].w+tr[u].size*x%p)%p+p)%p;
    }
    else
    {
        pushdown(u);
        int mid=(tr[u].l+tr[u].r)>>1;
        if(l<=mid) modify(u<<1,l,r,x);
        if(r>mid) modify(u<<1|1,l,r,x);
        pushup(u);
    }
}
int query(int u,int l,int r)
{
    if(l<=tr[u].l&&tr[u].r<=r) return tr[u].w;
    pushdown(u);
    int mid=(tr[u].l+tr[u].r)>>1,ans=0;
    if(l<=mid) ans=query(u<<1,l,r)%p;
    if(r>mid) ans=(ans+query(u<<1|1,l,r))%p;
    return ans;
}
void treesum(int x,int y)
{
    int ans=0;
    while(top[x]!=top[y])
    {

        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=(ans+query(1,idx[top[x]],idx[x]))%p;
        x=f[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans=(ans+query(1,idx[x],idx[y]))%p;
    printf("%d\n",ans);
}
void treeadd(int x,int y,int v)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        modify(1,idx[top[x]],idx[x],v);
        x=f[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    modify(1,idx[x],idx[y],v);
}
void solve()
{
    memset(h,-1,sizeof h);
    scanf("%d %d %d %d",&n,&m,&r,&p);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs1(r,0,1);
    dfs2(r,r);
    build(1,1,n);
    //cout<<query(1,1,n)<<endl;
    while(m--)
    {
        int op;
        scanf("%d",&op);
        if(op==1)
        {
            int x,y,v;
            scanf("%d %d %d",&x,&y,&v);
            treeadd(x,y,v);
        }
        else if(op==2)
        {
            int x,y;
            scanf("%d %d",&x,&y);
            treesum(x,y);
        }
        else if(op==3)
        {
            int x,v;
            scanf("%d %d",&x,&v);
            modify(1,idx[x],idx[x]+siz[x]-1,v);
        }
        else
        {
            int x;
            scanf("%d",&x);
            printf("%d\n",query(1,idx[x],idx[x]+siz[x]-1));
        }
    }
}
int main()
{
    int t=1;
    //scanf("%d",&t);
    while(t--)
    {
        solve();
    }
    return 0;
}