社区讨论

树剖模板过样例 但提交全wa求助

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

讨论操作

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

当前回复
3 条
当前快照
1 份
快照标识符
@lo7t5o12
此快照首次捕获于
2023/10/27 07:20
2 年前
此快照最后确认于
2023/10/27 07:20
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+5; 
int n,m,r,p,cnt,cntt;
int h[N],ne[N*2],e[N*2],pos[N];
int W[N],dfn[N],fa[N],son[N],top[N],V[N],dep[N],sz[N];
struct Node
{
	int l,r,sum,lz;
}T[N*10];
inline void add(int u,int v)
{
	ne[++cnt]=h[u];
	h[u]=cnt;
	e[cnt]=v;
}
int dfs1(int u,int f)
{
	sz[u]=1;
	fa[u]=f;
	dep[u]=dep[f]+1;
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int v=e[i];
		if(v == f) 		continue;
		sz[u]+=dfs1(v,u);
		if(sz[v]>sz[son[u]])
			son[u]=v;
	}
	return sz[u];
}
void dfs2(int u,int ttop)
{
	dfn[u]=++cntt;
	W[cntt]=V[u];
	top[u]=ttop;
	if(!son[u])		return ;
	dfs2(son[u],ttop);
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int v=e[i];
		if(v == fa[u] || v == son[u])	continue;
		dfs2(v,v);
	}
}
inline void pushdown(int rt)
{
	int lzz=T[rt].lz;
	if(lzz>0)
	{
		T[rt<<1].lz+=lzz%p;
		T[rt<<1].sum=(T[rt<<1].sum%p+(T[rt<<1].r-T[rt<<1].l+1)*T[rt<<1].lz%p)%p;
		T[rt<<1|1].lz+=lzz%p;
		T[rt<<1|1].sum=(T[rt<<1|1].sum%p+(T[rt<<1|1].r-T[rt<<1|1].l+1)*T[rt<<1|1].lz%p)%p;
		T[rt].lz=0;
	}
}
inline void pushup(int rt)
{
	T[rt].sum=(T[rt<<1].sum+T[rt<<1|1].sum)%p;
}
void modify(int l,int r,int k,int rt)
{
	k%=p;
	if(T[rt].l>=l && T[rt].r<=r)
	{
		T[rt].lz+=k;
		T[rt].sum+=(T[rt].r-T[rt].l+1)*k;
		return ;
	}
	pushdown(rt);
	int mid=T[rt].l+T[rt].r>>1;
	if(mid>=l)		modify(l,r,k,rt<<1);
	if(mid<r)	modify(l,r,k,rt<<1|1);
	pushup(rt);	
}
int query(int l,int r,int rt)
{
	if(T[rt].l>=l && T[rt].r<=r)
		return T[rt].sum%p;
	pushdown(rt);
	int s=0;
	int mid=T[rt].l+T[rt].r>>1;
	if(mid>=l)		s+=query(l,r,rt<<1);
	if(mid<r)		s+=query(l,r,rt<<1|1);	
	return s%p;
}
void build(int l,int r,int rt)
{
	T[rt]={l,r};
	if(l==r)
	{
		T[rt].sum=W[l]%p;
		pos[l]=rt;
		return ;
	}
	int mid=(l+r)>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	pushup(rt);
}
inline int qpath(int x,int y)
{
	int ans=0;
	while(top[x] != top[y])
	{
		if(dep[top[x]] < dep[top[y]])		swap(x,y);
		ans=ans%p+query(dfn[top[x]],dfn[x],1);
		x=fa[top[x]];
	}
	if(dep[x] > dep[y])		swap(x,y);
	ans=ans%p+query(dfn[x],dfn[y],1);
	return ans%p;
}
inline int qtree(int f)
{	
	return query(dfn[f],dfn[f]+sz[f]-1,1);
}
inline void updtree(int f,int k)
{
	k%=p;
	modify(dfn[f],dfn[f]+sz[f]-1,k,1);
}
inline void updpath(int x,int y,int k)
{
	k%=p;
	while(top[x] != top[y])
	{
		if(dep[top[x]] < dep[top[y]])	swap(x,y);
		modify(top[x],x,k,1);
		x=fa[top[x]];
	}
	if(dep[x] > dep[y])		swap(x,y);
	modify(dfn[x],dfn[y],k,1);
}
signed main()
{
	memset(h,-1,sizeof h);
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>m>>r>>p;
	for(int i=1;i<=n;i++)
	{
		cin>>V[i];
	} 
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		add(u,v);
		add(v,u);
	}
	dfs1(r,0);
	dfs2(r,r);
	build(1,n,1);
	for(int i=1;i<=m;i++)
	{
		int a,b,c,d;
		cin>>a;
		if(a==1)
		{
			cin>>b>>c>>d;
			updpath(b,c,d);
		}
		else if(a==2)
		{
			cin>>b>>c;
			cout<<qpath(b,c)<<endl;
		}
		else if(a==3)
		{
			cin>>b>>c;
			updtree(b,c);
		}
		else
		{
			cin>>b;
			cout<<qtree(b)<<endl;
		}
	}
	return 0;
}

回复

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

正在加载回复...