社区讨论

萌新求助~~20分

P3384【模板】重链剖分 / 树链剖分参与者 2已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mi7yn6t0
此快照首次捕获于
2025/11/21 05:45
4 个月前
此快照最后确认于
2025/11/21 05:45
4 个月前
查看原帖
CPP
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<algorithm>
#include<vector>
#include<sstream>
#include<cctype>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<string>
#include<ctime>
#define ll long long
#define ull unsigned long long
#define ff first
#define ss second
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
using namespace std;
const int INF=0x7f7f7f7f,mod=20100403;
inline int read()
{
	int x=0,k=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')k=-1;c=getchar();}
	while(c>='0'&&c<='9'){x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*k;
}
struct edge{
	int u,v,next;
}e[200086];
struct node{
	int laz,v;
}tree[400520];
int n,m,r,p,x,y,z,opt;
int d[200086],fa[200086],mson[200086],tot[200086],id[200086],w[200086],b[200086],cnt=0,top[200086];
int head[200086],num=0;
void store(int u,int v)
{
	e[++num].u=u,e[num].v=v;
	e[num].next=head[u];
	head[u]=num;
}
void dfs1(int x,int f,int dep)
{
	d[x]=dep;
	fa[x]=f;
	tot[x]=1;
	int maxsize=-1;
	for(int i=head[x];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==f)continue;
		dfs1(v,x,dep+1);
		tot[x]+=tot[v];
		if(tot[v]>maxsize)maxsize=tot[v],mson[x]=v;
	}
}
void dfs2(int x,int topm)
{
	id[x]=++cnt;
	w[cnt]=b[x];
	top[x]=topm;
	if(!mson[x])return ;
	dfs2(mson[x],topm);
	for(int i=head[x];i;i=e[i].next)
	{
		int v=e[i].v;
		if(!id[v])dfs2(v,v);
	}
}
void update(int root)
{
	tree[root].v=(tree[root<<1].v+tree[root<<1|1].v)%p;
}
void build(int root,int l,int r)
{
	if(l==r)
	{
		tree[root].v=w[l]%p;
		return ;
	}
	int mid=(l+r)>>1;
	build(root<<1,l,mid);
	build(root<<1|1,mid+1,r);
	update(root);
}
void pushdown(int root,int l,int r)
{
	int mid=(l+r)>>1;
	tree[root<<1].laz=(tree[root<<1].laz+tree[root].laz)%p;
	tree[root<<1|1].laz=(tree[root<<1|1].laz+tree[root].laz)%p;
	tree[root<<1].v=(tree[root<<1].v+tree[root].laz*(mid-l+1))%p;
	tree[root<<1|1].v=(tree[root<<1|1].v+tree[root].laz*(r-mid))%p;
	tree[root].laz=0;
}
void add(int root,int l,int r,int lnow,int rnow,int k)
{
	if(lnow>=l&&rnow<=r)
	{
		tree[root].laz=(tree[root].laz+k)%p;
		tree[root].v=(tree[root].v+k*(rnow-lnow+1))%p;
		return ;
	}
	pushdown(root,lnow,rnow);
	int mid=(lnow+rnow)>>1;
	if(l<=mid)add(root<<1,l,r,lnow,mid,k);
	if(r>mid)add(root<<1|1,l,r,mid+1,rnow,k);
	update(root);
}
int query(int root,int l,int r,int lnow,int rnow)
{
	if(lnow>=l&&rnow<=r)
	{
		return tree[root].v;
	}
	pushdown(root,lnow,rnow);
	int mid=(lnow+rnow)>>1,ans=0;
	if(l<=mid)ans=(ans+query(root<<1,l,r,lnow,mid))%p;
	if(r>mid)ans=(ans+query(root<<1|1,l,r,mid+1,rnow))%p;
	return ans;
}
int main()
{
	n=read(),m=read(),r=read(),p=read();
	rep(i,1,n)b[i]=read();
	rep(i,1,n-1)
	{
		x=read(),y=read();
		store(x,y);
		store(y,x);
	}
	dfs1(r,0,1);
	dfs2(r,r);
	build(1,1,n);
	while(m--)
	{
		opt=read();
		if(opt==1)
		{
			x=read(),y=read(),z=read();
			while(top[x]!=top[y])
			{
				if(d[top[x]]<d[top[y]])swap(x,y);
				add(1,id[top[x]],id[x],1,n,z);
				x=fa[top[x]];
			}
			if(d[top[x]]>d[top[y]])swap(x,y);
			add(1,id[x],id[y],1,n,z);
		}
		if(opt==2)
		{
			x=read(),y=read();
			int ans=0;
			while(top[x]!=top[y])
			{
				if(d[top[x]]<d[top[y]])swap(x,y);
				ans=(ans+query(1,id[top[x]],id[x],1,n))%p;
				x=fa[top[x]];
			}
			if(d[top[x]]>d[top[y]])swap(x,y);
			ans=(ans+query(1,id[x],id[y],1,n))%p;
			printf("%d\n",ans);
		}
		if(opt==3)
		{
			x=read(),z=read();
			add(1,id[x],id[x]+tot[x]-1,1,n,z);
		}
		if(opt==4)
		{
			x=read();
			int ans=query(1,id[x],id[x]+tot[x]-1,1,n);
			printf("%d\n",ans);
		}
	}
	return 0;
}

回复

1 条回复,欢迎继续交流。

正在加载回复...