树剖 RE 求调
查看原帖
树剖 RE 求调
290055
jr_inf楼主2023/4/14 21:26

rt,楼主眼睛不好,函数dfs1 RE

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
using namespace std;
const int N=2e5+10;
int n,m,r,p,w[N],cnt;
int sum[N*4];
int dep[N],fa[N],siz[N],son[N],top[N],id[N],nw[N];
vector<int> g[N];
void dfs1(int u,int f,int d)
{
	dep[u]=d;
	fa[u]=f;
	siz[u]=1;
	int maxs=-1;
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(v==f)continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if(siz[v]>maxs)
		{
			maxs=siz[v];
			son[u]=v;
		}
	}
	return;
}
void dfs2(int u,int t)
{
	id[u]=++cnt;
	nw[cnt]=w[u];
	top[u]=t;
	if(!son[u])return;
	dfs2(son[u],t);
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(v==fa[u]||v==son[u])continue;
		dfs2(v,v);
	}
	return;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void build(int o,int l,int r)
{
	if(l==r)
	{
		sum[o]=nw[l];
		return;
	}
	int mid=(l+r)>>1;
	build(o*2,l,mid);
	build(o*2+1,mid+1,r);
	sum[o]=(sum[o*2]+sum[o*2+1])%p;
	return;
}
void update(int o,int l,int r,int x,int y,int z)
{
	if(r<x||l>y)return;
	if(l==r)
	{
		sum[o]=(sum[o]+z+p)%p;
		return;
	}
	int mid=l+r>>1;
	update(o*2,l,mid,x,y,z);
	update(o*2+1,mid+1,r,x,y,z);
	sum[o]=sum[o*2]+sum[o*2+1];
	return;
}
int query(int o,int l,int r,int x,int y)
{
	if(r<x||l>y)return 0;
	if(x<=l&&r<=y)return sum[o];
	int mid=l+r>>1;
	return (query(o*2,l,mid,x,y)+query(o*2+1,mid+1,r,x,y))%p;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void update2(int x,int y,int z)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		update(1,1,n,id[top[x]],id[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	update(1,1,n,id[x],id[y],z);
	return;
}
int query2(int x,int y)
{
	int ans=0,cnt;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		cnt=query(1,1,n,id[top[x]],id[x]);
		ans=(ans+cnt)%p;
		x=fa[top[x]];
	}
	if(dep[top[x]]>dep[top[y]])swap(x,y);
	cnt=query(1,1,n,id[x],id[y]);
	ans=(ans+cnt)%p;
	return ans;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
void pr()
{
	for(int i=1;i<=n;i++)printf("%d ",query(1,1,n,id[i],id[i]));
	printf("\n");
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&r,&p);
	for(int i=1;i<=n;i++)scanf("%d",&w[i]);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		g[x].push_back(y);
		g[y].push_back(x);
	}
	dfs1(r,0,1);
	dfs2(r,r);
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int opt,x,y,z;
		scanf("%d",&opt);
		if(opt==1)
		{
			scanf("%d%d%d",&x,&y,&z);
			update2(x,y,z%p);
//			pr();//
		}
		if(opt==2)
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",query2(x,y));
		}
		if(opt==3)
		{
			scanf("%d%d",&x,&z);
			update(1,1,n,id[x],id[x]+siz[x],z%p);
//			pr();//
		}
		if(opt==4)
		{
			scanf("%d",&x);
			printf("%d\n",query(1,1,n,id[x],id[x]+siz[x]-1));
		}
	}
}
2023/4/14 21:26
加载中...