树剖 RE 求调

rt，楼主眼睛不好，函数dfs1 RE

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
using namespace std;
const int N=2e5+10;
int n,m,r,p,w[N],cnt;
int sum[N*4];
int dep[N],fa[N],siz[N],son[N],top[N],id[N],nw[N];
vector<int> g[N];
void dfs1(int u,int f,int d)
{
    dep[u]=d;
    fa[u]=f;
    siz[u]=1;
    int maxs=-1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==f)continue;
        dfs1(v,u,d+1);
        siz[u]+=siz[v];
        if(siz[v]>maxs)
        {
            maxs=siz[v];
            son[u]=v;
        }
    }
    return;
}
void dfs2(int u,int t)
{
    id[u]=++cnt;
    nw[cnt]=w[u];
    top[u]=t;
    if(!son[u])return;
    dfs2(son[u],t);
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa[u]||v==son[u])continue;
        dfs2(v,v);
    }
    return;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void build(int o,int l,int r)
{
    if(l==r)
    {
        sum[o]=nw[l];
        return;
    }
    int mid=(l+r)>>1;
    build(o*2,l,mid);
    build(o*2+1,mid+1,r);
    sum[o]=(sum[o*2]+sum[o*2+1])%p;
    return;
}
void update(int o,int l,int r,int x,int y,int z)
{
    if(r<x||l>y)return;
    if(l==r)
    {
        sum[o]=(sum[o]+z+p)%p;
        return;
    }
    int mid=l+r>>1;
    update(o*2,l,mid,x,y,z);
    update(o*2+1,mid+1,r,x,y,z);
    sum[o]=sum[o*2]+sum[o*2+1];
    return;
}
int query(int o,int l,int r,int x,int y)
{
    if(r<x||l>y)return 0;
    if(x<=l&&r<=y)return sum[o];
    int mid=l+r>>1;
    return (query(o*2,l,mid,x,y)+query(o*2+1,mid+1,r,x,y))%p;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void update2(int x,int y,int z)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        update(1,1,n,id[top[x]],id[x],z);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    update(1,1,n,id[x],id[y],z);
    return;
}
int query2(int x,int y)
{
    int ans=0,cnt;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        cnt=query(1,1,n,id[top[x]],id[x]);
        ans=(ans+cnt)%p;
        x=fa[top[x]];
    }
    if(dep[top[x]]>dep[top[y]])swap(x,y);
    cnt=query(1,1,n,id[x],id[y]);
    ans=(ans+cnt)%p;
    return ans;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void pr()
{
    for(int i=1;i<=n;i++)printf("%d ",query(1,1,n,id[i],id[i]));
    printf("\n");
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&r,&p);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs1(r,0,1);
    dfs2(r,r);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int opt,x,y,z;
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%d",&x,&y,&z);
            update2(x,y,z%p);
//          pr();//
        }
        if(opt==2)
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",query2(x,y));
        }
        if(opt==3)
        {
            scanf("%d%d",&x,&z);
            update(1,1,n,id[x],id[x]+siz[x],z%p);
//          pr();//
        }
        if(opt==4)
        {
            scanf("%d",&x);
            printf("%d\n",query(1,1,n,id[x],id[x]+siz[x]-1));
        }
    }
}