#include<bits/stdc++.h>
using namespace std;
int n,m,d,mod,w[200005],wt[200005],e[200005],ne[200005],h[200005],idx=1,t[800005],l[800005],r[800005],tag[800005];
int son[200005],id[200005],fa[200005],cnt,dep[200005],siz[200005],top[200005];
/*son[]重儿子,id[]新编号,fa[]父亲节点,cnt dfs序,dep[]深度,siz[]子树大小,top[]当前链顶端节点*/
int k,a,b,c;
inline void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;}
void dfs1(int u,int f,int de)
{
dep[u]=de,fa[u]=f,siz[u]=1;int mson=-1;
for(int i=h[u];i;i=ne[i])
if(e[i]!=f)
{
dfs1(e[i],u,de+1),siz[u]+=siz[e[i]];
if(siz[e[i]]>mson) son[u]=e[i],mson=siz[e[i]];
}
}
void dfs2(int u,int to)
{
//cout<<u<<" ";
id[u]=++cnt,wt[cnt]=w[u],top[u]=to;
if(!son[u]) return;
dfs2(son[u],to);
for(int i=h[u];i;i=ne[i]) if(e[i]!=fa[u]&&e[i]!=son[u]) dfs2(e[i],e[i]);
}
inline void pushup(int p){t[p]=(t[p<<1]+t[p<<1|1])%mod;}
inline void addtag(int p,int k){t[p]=(t[p]+(r[p]-l[p]+1)*k)%mod,tag[p]=(tag[p]+k)%mod;}
inline void pushdown(int p){addtag(p<<1,tag[p]),addtag(p<<1|1,tag[p]),tag[p]=0;}
void build(int p,int pl,int pr)
{
l[p]=pl,r[p]=pr;
if(pl==pr){t[p]=wt[pl]%mod;return;}
int mid=pl+pr>>1;
build(p<<1,pl,mid),build(p<<1|1,mid+1,pr);
pushup(p);
}
void update(int p,int pl,int pr,int k)
{
if(l[p]>=pl&&r[p]<=pr){addtag(p,k);return;}
if(tag[p]) pushdown(p);
int mid=l[p]+r[p]>>1;
if(mid>=pl) update(p<<1,pl,pr,k);
if(mid<pr) update(p<<1|1,pl,pr,k);
pushup(p);
}
int ask(int p,int pl,int pr)
{
if(l[p]>=pl&&r[p]<=pr) return t[p]%mod;
if(tag[p]) pushdown(p);
int mid=l[p]+r[p]>>1,ans=0;
if(mid>=pl) ans=ask(p<<1,pl,pr)%mod;
if(mid<pr) ans=(ans+ask(p<<1|1,pl,pr))%mod;
return ans%mod;
}
inline int ask_r(int x,int y)
{
int ans=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans=(ans+ask(1,id[top[x]],id[x]))%mod,x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
return (ans+ask(1,id[x],id[y]))%mod;
}
inline void update_r(int x,int y,int k)
{
k%=mod;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
update(1,id[top[x]],id[x],k),x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
update(1,id[x],id[y],k);
}
inline int ask_o(int x)
{
return ask(1,id[x],id[x]+siz[x]-1)%mod;
}
inline void update_o(int x,int k)
{
update(1,id[x],id[x]+siz[x]-1,k%mod);
}
int main()
{
scanf("%d%d%d%d",&n,&m,&d,&mod);
for(int i=1;i<=n;++i) scanf("%d",&w[i]);
for(int i=1;i<n;++i) scanf("%d%d",&a,&b),add(a,b),add(b,a);
dfs1(d,0,1),dfs2(d,d),build(1,1,n);
//for(int i=1;i<=n;i++) cout<<son[i]<<" ";puts("");
while(m--)
{
scanf("%d",&k);
if(k==1) scanf("%d%d%d",&a,&b,&c),update_r(a,b,c);
else if(k==2) scanf("%d%d",&a,&b),printf("%d\n",ask_r(a,b));
else if(k==3) scanf("%d%d",&a,&b),update_o(a,b);
else scanf("%d",&a),printf("%d\n",ask_o(a));
}
return 0;
}
树链剖分/重链剖分模板
于 2023-09-08 08:30:14 首次发布