P3384 【模板】轻重链剖分

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e6+5;
int n,m,r,p,tot,cnt;
int val[maxn],head[maxn],dep[maxn],top[maxn],id[maxn],re_val[maxn],sz[maxn],fa[maxn],son[maxn];
struct tree
{
	int l,r,sz,w,lz;
}tr[maxn<<2];
struct edge
{
	int to,nxt;
}e[maxn<<1];
void add(int x,int y)
{
	e[++tot].to=y;
	e[tot].nxt=head[x];
	head[x]=tot;
}
void dfs1(int u,int f,int step)
{
	dep[u]=step;
	fa[u]=f;
	sz[u]=1;
	int maxson=-1;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==f) continue;
		dfs1(to,u,step+1);
		sz[u]+=sz[to];
		if(sz[to]>maxson)
		{
			maxson=sz[to];
			son[u]=to;
		}
	}
}
void dfs2(int u,int roof)
{
	id[u]=++cnt;
	re_val[cnt]=val[u];
	top[u]=roof;
	if(!son[u]) return;
	dfs2(son[u],roof);
	for(int i=head[u];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(!id[to])
			dfs2(to,to);
	}
}
void pushup(int now)
{
	tr[now].w=(tr[now<<1].w+tr[now<<1|1].w+p)%p;
}
void pushdown(int now)
{
	if(!tr[now].lz) return;
	tr[now<<1].w=(tr[now<<1].w+tr[now<<1].sz*tr[now].lz)%p;
	tr[now<<1|1].w=(tr[now<<1|1].w+tr[now<<1|1].sz*tr[now].lz)%p;
	tr[now<<1].lz=(tr[now<<1].lz+tr[now].lz)%p;
	tr[now<<1|1].lz=(tr[now<<1|1].lz+tr[now].lz)%p;
	tr[now].lz=0;
}
void build(int now,int l,int r)
{
	tr[now].sz=r-l+1;
	tr[now].l=l; tr[now].r=r;
	if(l==r)
	{
		tr[now].w=re_val[l];
		return;	
	}
	int mid=(l+r)>>1;
	build(now<<1,l,mid);
	build(now<<1|1,mid+1,r);
	pushup(now);
}
void tr_add(int now,int l,int r,int z)
{
	if(l<=tr[now].l && tr[now].r<=r)
	{
		tr[now].w+=z*tr[now].sz;
		tr[now].lz+=z;
		return;
	}
	pushdown(now);
	int mid=(tr[now].l+tr[now].r)>>1;
	if(l<=mid) tr_add(now<<1,l,r,z);
	if(mid<r) tr_add(now<<1|1,l,r,z);
	pushup(now);
}
void tree_add(int x,int y,int z)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		tr_add(1,id[top[x]],id[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	tr_add(1,id[x],id[y],z);
}
int tr_calc(int now,int l,int r)
{
	int ans=0;
	if(l<=tr[now].l && tr[now].r<=r)
	{
		return tr[now].w;
	}
	pushdown(now);
	int mid=(tr[now].l+tr[now].r)>>1;
	if(l<=mid) ans=(ans+tr_calc(now<<1,l,r))%p;
	if(mid<r) ans=(ans+tr_calc(now<<1|1,l,r))%p;
	return ans;
}
void query(int x,int y)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ans=(ans+tr_calc(1,id[top[x]],id[x]))%p;
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans=(ans+tr_calc(1,id[x],id[y]))%p;
	printf("%d\n",ans%p);
}
int main()
{

	scanf("%d%d%d%d",&n,&m,&r,&p);
	int x,y;
	for(int i=1;i<=n;i++) scanf("%d",&val[i]);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dfs1(r,0,1);
	dfs2(r,r);
	build(1,1,n);
	int op,z;
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d%d",&x,&y,&z);
			z%=p;
			tree_add(x,y,z);
		}
		if(op==2)
		{
			scanf("%d%d",&x,&y);
			query(x,y);
		}
		if(op==3)
		{
			scanf("%d%d",&x,&y); y%=p;
			tr_add(1,id[x],id[x]+sz[x]-1,y);
		}
		if(op==4)
		{
			scanf("%d",&x);
			printf("%d\n",tr_calc(1,id[x],id[x]+sz[x]-1)%p);
		}
	}
	return 0;
}