树链剖分/重链剖分模板

去洛谷看我的博客

#include<bits/stdc++.h>
using namespace std;
int n,m,d,mod,w[200005],wt[200005],e[200005],ne[200005],h[200005],idx=1,t[800005],l[800005],r[800005],tag[800005];
int son[200005],id[200005],fa[200005],cnt,dep[200005],siz[200005],top[200005];
/*son[]重儿子，id[]新编号，fa[]父亲节点，cnt dfs序，dep[]深度，siz[]子树大小，top[]当前链顶端节点*/
int k,a,b,c;
inline void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;}
void dfs1(int u,int f,int de)
{
	dep[u]=de,fa[u]=f,siz[u]=1;int mson=-1;
	for(int i=h[u];i;i=ne[i])
		if(e[i]!=f)
		{
			dfs1(e[i],u,de+1),siz[u]+=siz[e[i]];
			if(siz[e[i]]>mson) son[u]=e[i],mson=siz[e[i]];
		}
}
void dfs2(int u,int to)
{
	//cout<<u<<" ";
	id[u]=++cnt,wt[cnt]=w[u],top[u]=to;
	if(!son[u]) return;
	dfs2(son[u],to);
	for(int i=h[u];i;i=ne[i]) if(e[i]!=fa[u]&&e[i]!=son[u]) dfs2(e[i],e[i]);
}
inline void pushup(int p){t[p]=(t[p<<1]+t[p<<1|1])%mod;}
inline void addtag(int p,int k){t[p]=(t[p]+(r[p]-l[p]+1)*k)%mod,tag[p]=(tag[p]+k)%mod;}
inline void pushdown(int p){addtag(p<<1,tag[p]),addtag(p<<1|1,tag[p]),tag[p]=0;}
void build(int p,int pl,int pr)
{
	l[p]=pl,r[p]=pr;
	if(pl==pr){t[p]=wt[pl]%mod;return;}
	int mid=pl+pr>>1;
	build(p<<1,pl,mid),build(p<<1|1,mid+1,pr);
	pushup(p);
}
void update(int p,int pl,int pr,int k)
{
	if(l[p]>=pl&&r[p]<=pr){addtag(p,k);return;}
	if(tag[p]) pushdown(p);
	int mid=l[p]+r[p]>>1;
	if(mid>=pl) update(p<<1,pl,pr,k);
	if(mid<pr) update(p<<1|1,pl,pr,k);
	pushup(p);
}
int ask(int p,int pl,int pr)
{
	if(l[p]>=pl&&r[p]<=pr) return t[p]%mod;
	if(tag[p]) pushdown(p);
	int mid=l[p]+r[p]>>1,ans=0;
	if(mid>=pl) ans=ask(p<<1,pl,pr)%mod;
	if(mid<pr) ans=(ans+ask(p<<1|1,pl,pr))%mod;
	return ans%mod;
}
inline int ask_r(int x,int y)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ans=(ans+ask(1,id[top[x]],id[x]))%mod,x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	return (ans+ask(1,id[x],id[y]))%mod;
}
inline void update_r(int x,int y,int k)
{
	k%=mod;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		update(1,id[top[x]],id[x],k),x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update(1,id[x],id[y],k);
}
inline int ask_o(int x)
{
	return ask(1,id[x],id[x]+siz[x]-1)%mod;
}
inline void update_o(int x,int k)
{
	update(1,id[x],id[x]+siz[x]-1,k%mod);
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&d,&mod);
	for(int i=1;i<=n;++i) scanf("%d",&w[i]);
	for(int i=1;i<n;++i) scanf("%d%d",&a,&b),add(a,b),add(b,a);
	dfs1(d,0,1),dfs2(d,d),build(1,1,n);
	//for(int i=1;i<=n;i++) cout<<son[i]<<" ";puts("");
	while(m--)
	{
		scanf("%d",&k);
		if(k==1) scanf("%d%d%d",&a,&b,&c),update_r(a,b,c);
		else if(k==2) scanf("%d%d",&a,&b),printf("%d\n",ask_r(a,b));
		else if(k==3) scanf("%d%d",&a,&b),update_o(a,b);
		else scanf("%d",&a),printf("%d\n",ask_o(a));
	}
	return 0;
}