蒟蒻10pts求调教
查看原帖
蒟蒻10pts求调教
754502
_AyachiNene楼主2023/5/10 20:43
#include<bits/stdc++.h>
#define maxn 114514
#define ls root*2
#define rs root*2+1
using namespace std;
struct node
{
	int nxt,to;
}e[maxn*2];
//----------线段树-------------------
int k;
int n,m,r,p; 

struct node1
{
	int l,r,val,f;
}t[maxn*4];
void bld(int l,int r,int root)
{
	t[root].l=l;
	t[root].r=r;
	if(l==r)
	{
		cin>>t[root].val;
		t[root].val%=p;
		return;
	}
	int mid=(l+r)/2;
	bld(l,mid,root*2);
	bld(mid+1,r,root*2+1);
	t[root].val=(t[ls].val+t[rs].val)%p;
}
void down(int root)
{
	t[ls].f+=t[root].f;
	t[rs].f+=t[root].f;
	t[ls].val=(t[ls].val+(t[ls].r-t[ls].l+1)*t[root].f)%p;
	t[rs].val=(t[rs].val+(t[rs].r-t[rs].l+1)*t[root].f)%p;
	t[root].f=0;
}
void add2(int x,int y,int root,int k)
{
	if(t[root].l>=x&&t[root].r<=y)
	{
		t[root].val+=((t[root].r-t[root].l+1)*k)%p;
		t[root].f+=k;
		return; 
	}
	if(t[root].f)
		down(root);
	int mid=(t[root].l+t[root].r)/2;
	if(x<=mid)
		add2(x,y,root*2,k);
	if(y>mid)
		add2(x,y,root*2+1,k);
	t[root].val=(t[ls].val+t[rs].val)%p;
}
int query(int x,int y,int root)
{
	int ret=0;
	if(t[root].l>=x&&t[root].r<=y)
		return t[root].val;
	if(t[root].f)
		down(root);
	int mid=(t[root].l+t[root].r)/2;
	if(x<=mid)
		ret=(ret+query(x,y,ls))%p;
	if(y>mid)
		ret=(ret+query(x,y,rs))%p;
	return ret;
}
//----------------------------------- 
int head[maxn*2],cnt1;
void add(int u,int v)
{
	e[++cnt1].to=v;
	e[cnt1].nxt=head[u];
	head[u]=cnt1;
}
int top[maxn],dfn[maxn],rk[maxn],son[maxn],size[maxn],f[maxn],dep[maxn],cnt;
void dfs1(int u,int fa)
{
	size[u]=1;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to;
		if(v!=fa)
		{
			dep[v]=dep[u]+1;
			f[v]=u;
			dfs1(v,u);
			size[u]+=size[v];
			if(size[v]>size[son[u]])
				son[u]=v;
		}
	}
}
void dfs2(int u,int t)
{
	dfn[++cnt]=u;
	top[u]=t;
	if(son[u])
		dfs2(son[u],t);
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to;
		if(v!=f[u]&&v!=son[u])
			dfs2(v,v);
	}
}
int sum(int x,int y)
{
	int ans=0,fx=top[x],fy=top[y];
	while(fx!=fy)
	{
		if(dep[fx]<dep[fy])
			swap(x,y),swap(fx,fy);
		
		ans+=query(dfn[fx],dfn[x],1);;
		x=f[fx],fx=top[x];
	}
	if(dfn[x]>dfn[y])
		swap(x,y);
	ans+=query(dfn[x],dfn[y],1);
	return ans;
}
void add1(int x,int y,int z)
{
    int fx=top[x],fy=top[y];
    while(fx!=fy)
	{
        if(dep[fx]<dep[fy])
			swap(x,y),swap(fx,fy);
        add2(dfn[fx],dfn[x],1,z);
        x=f[fx],fx=top[x];
    }
    if(dfn[x]>dfn[y])
		swap(x,y);
    add2(dfn[x],dfn[y],1,z);
}
int main()
{
	cin>>n>>m>>r>>p;
	bld(1,n,1);
	for(int i=1;i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	dfs1(r,0);
	dfs2(r,r);
	while(m--)
	{
		int op,x,y,z;
		cin>>op;
		if(op==1)
		{
			cin>>x>>y>>z;
			add1(x,y,z);
		}
		else if(op==2)
		{
			cin>>x>>y;
			k=0;
			cout<<sum(x,y)<<endl;
		}
		else if(op==3)
		{
			cin>>x>>z;
			add2(dfn[x],dfn[x]+size[x]-1,1,z);
		}
		else
		{
			cin>>x;
			cout<<query(dfn[x],dfn[x]+size[x]-1,1)<<endl;
		}
	}
//	for(int i=1;i<=10;i++)
//		cout<<t[i].val<<' ';
}
2023/5/10 20:43
加载中...