社区讨论

树剖求救,虽然我觉得不会有人回复我QwQ

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

讨论操作

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

当前回复
4 条
当前快照
1 份
快照标识符
@mi7yhmgl
此快照首次捕获于
2025/11/21 05:41
4 个月前
此快照最后确认于
2025/11/21 05:41
4 个月前
查看原帖
CPP
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cmath>
#define int long long
using namespace std; 

const int N=5000005;

int e=0,ver[N<<1],nexts[N<<1],head[N];
void add(int x,int y){ver[++e]=y,nexts[e]=head[x],head[x]=e;}

struct Tc{int dep,fa,size,son,id,top;}Tcut[N];
int cnt;

struct Tree{int dat,add,l,r;}t[N<<2];

int val[N],tval[N];

int n,m,root,mol;

void dfs1(int x,int fa,int deep)
{
	Tcut[x].size=1;
	Tcut[x].dep=deep;
	Tcut[x].fa=fa;
	int Maxson=-1; 
	for(int i=head[x];i;i=nexts[i])
	{
		int v=ver[i];
		if(v==fa)continue;
		dfs1(v,x,deep+1);
		Tcut[x].size+=Tcut[v].size;
		if(Maxson<Tcut[v].size){Tcut[x].son=v;	Maxson=Tcut[v].size;}
	}
}

void dfs2(int x,int top)
{
	Tcut[x].id=++cnt;
	tval[cnt]=val[x]%mol;
	Tcut[x].top=top;
	if(!Tcut[x].son)return;
	dfs2(Tcut[x].son,top);
	for(int i=head[x];i;i=nexts[i])
	{
		int v=ver[i];
		if(v==Tcut[x].fa||v==Tcut[x].son)continue;
		dfs2(v,v);
	}
}

void pushup(int p){t[p].dat=t[p<<1].dat+t[p<<1|1].dat;t[p].dat%=mol;}

void build(int p,int l,int r)
{
	t[p].l=l;t[p].r=r;
	if(l==r){t[p].dat=tval[l];t[p].dat%=mol;return;}
	int mid=l+r>>1;
	build(p<<1,l,mid);build(p<<1|1,mid+1,r);
	pushup(p);
}

void pushdown(int p)
{
	if(t[p].add)
	{
		t[p<<1].add+=t[p].add;t[p<<1|1].add+=t[p].add;
		t[p<<1].dat+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
		t[p<<1|1].dat+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1); 
		t[p<<1].add%=mol,t[p<<1|1].add%=mol,t[p<<1].dat%=mol,t[p<<1|1].dat%=mol;
		t[p].add=0;
	}
}

int query(int p,int l,int r)
{
	if(t[p].l>=l&&r>=t[p].r)return t[p].dat;
	pushdown(p);
	int mid=t[p].l+t[p].r>>1;
	int ans=0;
	if(l<=mid)ans+=query(p<<1,l,r),ans%=mol;
	if(mid<r)ans+=query(p<<1|1,l,r),ans%=mol;
	return ans%mol;
}

void change(int p,int l,int r,int v)
{
	if(l<=t[p].l&&r>=t[p].r)
	{
		t[p].add+=v;
		t[p].dat+=v*(t[p].r-t[p].l+1);
		t[p].add%=mol;
		t[p].dat%=mol;
		return;
	}
	int mid=t[p].l+t[p].r>>1;
	pushdown(p);
	if(mid>=l)change(p<<1,l,r,v);
	if(r>mid)change(p<<1|1,l,r,v);
	pushup(p);
}

int query_edge(int x,int y)
{
	int ans=0;
	while(Tcut[x].top!=Tcut[y].top)
	{
		if(Tcut[x].dep<Tcut[y].dep)swap(x,y);
		ans+=query(1,Tcut[x].id,Tcut[Tcut[x].top].id);
		ans%=mol;
		x=Tcut[Tcut[x].top].fa;
	}
	if(Tcut[x].dep>Tcut[y].dep)swap(x,y);
	ans+=query(1,Tcut[x].id,Tcut[y].id);
	return ans%mol;
}

void change_edge(int x,int y,int z)
{
	z%=mol;
	while(Tcut[x].top!=Tcut[y].top)
	{
		if(Tcut[x].top<Tcut[y].top)swap(x,y);
		change(1,Tcut[x].id,Tcut[Tcut[x].top].id,z);
		x=Tcut[Tcut[x].top].fa;
	}
	if(Tcut[x].dep>Tcut[y].dep)swap(x,y);
	change(1,Tcut[x].id,Tcut[y].id,z);
} 

int query_p(int x)
{
	return query(1,Tcut[x].id,Tcut[x].id+Tcut[x].size-1)%mol;
}

void change_p(int x,int z)
{
	change(1,Tcut[x].id,Tcut[x].id+Tcut[x].size-1,z); 
}

signed main()
{
	scanf("%lld%lld%lld%lld",&n,&m,&root,&mol);
	for(int i=1;i<=n;i++)scanf("%lld",&val[i]);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%lld%lld",&x,&y);
		add(x,y);add(y,x);
	}
	dfs1(root,0,1);dfs2(root,root);build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int k,x,y,z;
		scanf("%lld",&k);
		if(k==1){scanf("%lld%lld%lld",&x,&y,&z);change_edge(x,y,z);}
		else if(k==2){scanf("%lld%lld",&x,&y);printf("%lld\n",query_edge(x,y)%mol);}
		else if(k==3){scanf("%lld%lld",&x,&z);change_p(x,z);}
		else scanf("%lld",&x),printf("%lld\n",query_p(x)%mol);
	}
	return 0;
}

回复

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

正在加载回复...