沃日,调了半天
代码:
#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));
}
}
}