树链模板

#include <cstdio>
#define ls id<<1
#define rs id<<1|1
#define mid (L[id]+R[id]>>1)
const int N=100010;
int max(int x,int y){return x>y?x:y;}
int min(int x,int y){return x<y?x:y;}
int dep[N],siz[N],ws[N],dfn[N],ha[N],top[N],f[N],w[N],tot=0;
int n,m,p,r;
int to[N<<1],next[N<<1],head[N],cnt=0;
void add(int u,int v)
{
    to[++cnt]=v;next[cnt]=head[u];head[u]=cnt;
}
void dfs0(int now)
{
    for(int i=head[now];i;i=next[i])
    {
        int v=to[i];
        if(v!=f[now])
        {
            dep[v]=dep[now]+1;
            f[v]=now;
            dfs0(v);
            siz[now]+=siz[v];
            if(siz[ws[now]]<siz[v])
                ws[now]=v;
        }
    }
    siz[now]++;
}
void dfs1(int now,int anc)
{
    dfn[now]=++tot;
    ha[tot]=now;
    top[now]=anc;
    if(!ws[now]) return;
    dfs1(ws[now],anc);
    for(int i=head[now];i;i=next[i])
    {
        int v=to[i];
        if(!dfn[v])
            dfs1(v,v);
    }
}
int L[N<<2],R[N<<2],dat[N<<2],lazy[N<<2];
void build(int id,int l,int r)
{
    L[id]=l,R[id]=r;
    if(l==r) {dat[id]=w[ha[l]];return;}
    build(ls,l,mid);
    build(rs,mid+1,r);
    dat[id]=(dat[ls]+dat[rs])%p;
}
void change(int id,int l,int r,int delta)
{
    if(L[id]==l&&R[id]==r)
    {
        lazy[id]=(lazy[id]+delta)%p;
        return;
    }
    dat[id]=(dat[id]+delta%p*(r+1-l)%p)%p;
    if(r<=mid) change(ls,l,r,delta);
    else if(l>mid) change(rs,l,r,delta);
    else change(ls,l,mid,delta),change(rs,mid+1,r,delta);
}
void push_down(int id)
{
    dat[id]=(lazy[id]%p*(R[id]-L[id]+1)%p+dat[id])%p;
    if(L[id]!=R[id])
    {
        lazy[ls]+=lazy[id];
        lazy[rs]+=lazy[id];
    }
    lazy[id]=0;
}
int query(int id,int l,int r)
{
    push_down(id);
    if(L[id]==l&&R[id]==r)
        return dat[id];
    if(r<=mid) return query(ls,l,r)%p;
    else if(l>mid) return query(rs,l,r)%p;
    else return (query(ls,l,mid)+query(rs,mid+1,r))%p;
}
void t_get(int x,int y)
{
    int ans=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]>dep[top[y]])
        {
            ans=(ans+query(1,dfn[top[x]],dfn[x]))%p;
            x=f[top[x]];
        }
        else
        {
            ans=(ans+query(1,dfn[top[y]],dfn[y]))%p;
            y=f[top[y]];
        }
    }
    ans=(ans+query(1,min(dfn[x],dfn[y]),max(dfn[x],dfn[y])))%p;
    printf("%d
",ans);
}
void t_add(int x,int y,int delta)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]>dep[top[y]])
        {
            change(1,dfn[top[x]],dfn[x],delta);
            x=f[top[x]];
        }
        else
        {
            change(1,dfn[top[y]],dfn[y],delta);
            y=f[top[y]];
        }
    }
    change(1,min(dfn[x],dfn[y]),max(dfn[x],dfn[y]),delta);
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&r,&p);
    for(int i=1;i<=n;i++)
        scanf("%d",w+i);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    dfs0(r);
    dfs1(r,r);
    build(1,1,n);
    int x,y,z,opt;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%d",&x,&y,&z);
            t_add(x,y,z);
        }
        else if(opt==2)
        {
            scanf("%d%d",&x,&y);
            t_get(x,y);
        }
        else if(opt==3)
        {
            scanf("%d%d",&x,&z);
            change(1,dfn[x],dfn[x]+siz[x]-1,z);
        }
        else
        {
            scanf("%d",&x);
            printf("%d
",query(1,dfn[x],dfn[x]+siz[x]-1));
        }
    }
    return 0;
}