树链剖分板子

https://www.luogu.com.cn/problem/P3384e



 

 关于具体的思想可以看这个up的视频，做的很好:  https://www.bilibili.com/video/BV1Qt411u77f?spm_id_from=333.999.0.0

对于操作3和操作4直接利用dfs序,因为dfs序各个子树他们的子树的节点是连续的,所以可以直接用线段树进行修改，对于操作1或者操作2，如果是单独出现的可以通过树上差分解决,但是如果在询问里面1，2操作都有，那么每次进行2询问前都要dfs一遍，显然会超时，所以树链剖分华丽登场了

对于操作1和操作2，可以先进行重链和轻链的划分(子节点多的就是重儿子)，然后记录一下各个节点的父节点是什么，以及各个节点对应的重链的根节点,然后对于x和y可以不断向其所在重链的根节点跳跃，在跳跃的过程中dfn[top[x]]和dfn[x]一定是连续的把这个答案加上,然后再把x换位这个重链的根节点的父节点，即x=pre[top[x]] 进行同样操作，一定可以跳到同一条重链上面,解决

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=2e6+5;
vector<int>vec[maxn];
int n,m,r,mod;
int v[maxn];
int pre[maxn],depth[maxn],siz[maxn],son[maxn],b[maxn];
void dfs1(int u,int fa){
	pre[u]=fa;
	depth[u]=depth[fa]+1;
	siz[u]=1;
	int maxsize=-1;
	for(int v:vec[u]){
		if(v==fa) continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxsize){
			maxsize=siz[v];
			son[u]=v;
		}
	}
}
int tim,dfn[maxn],top[maxn],w[maxn];
void dfs2(int u,int t){
	dfn[u]=++tim;
	top[u]=t;
	w[tim]=v[u];
	if(!son[u]){
		return;
	}
	dfs2(son[u],t);
	for(int v:vec[u]){
		if(v==pre[u]||v==son[u]){
			continue;
		}
		dfs2(v,v);
	}
}
int d[maxn];
void build(int l,int r,int p){
	if(l==r){
		d[p]=w[l]%mod;
		return;
	}
	int m=(l+r)>>1;
	build(l,m,p*2);
	build(m+1,r,p*2+1);
	d[p]=(d[p*2]+d[p*2+1])%mod;
}
void modify(int l,int r,int s,int t,int x,int p){
	if(s>=l&&t<=r){
		d[p]+=(t-s+1)*x;
		d[p]%=mod;
		b[p]=(b[p]+x)%mod;
		return;
	}
	int m=(s+t)>>1;
	if(b[p]){
		d[p*2]=(d[p*2]+(m-s+1)*b[p])%mod,d[p*2+1]=(d[p*2+1]+(t-m)*b[p])%mod;
		b[p*2]=(b[p*2]+b[p])%mod,b[p*2+1]=(b[p*2+1]+b[p])%mod,b[p]=0;
	}
	if(m>=l){
		modify(l,r,s,m,x,p*2);
	}
	if(m+1<=r){
		modify(l,r,m+1,t,x,p*2+1);
	}
	d[p]=(d[p*2]+d[p*2+1])%mod;
}
int query(int l,int r,int s,int t,int p){
	if(s>=l&&t<=r){
		return d[p];
	}
	int m=(s+t)>>1;
	if(b[p]){
		d[p*2]=(d[p*2]+(m-s+1)*b[p])%mod,d[p*2+1]=(d[p*2+1]+(t-m)*b[p])%mod;
		b[p*2]=(b[p*2]+b[p])%mod,b[p*2+1]=(b[p*2+1]+b[p])%mod,b[p]=0;
	}
	int sum=0;
	if(l<=m){
		sum=(sum+query(l,r,s,m,p*2))%mod;
	}
	if(m+1<=r){
		sum=(sum+query(l,r,m+1,t,p*2+1))%mod;
	}
	return sum;
}
void mson(int x,int z){  //修改子树
	modify(dfn[x],dfn[x]+siz[x]-1,1,n,z,1);
} 
int qson(int x){   //查询子树
	return query(dfn[x],dfn[x]+siz[x]-1,1,n,1);
}
void mchain(int x,int y,int z){    //修改一条链
	while(top[x]!=top[y]){
		if(depth[top[x]]<depth[top[y]]){
			swap(x,y);
		}
		modify(dfn[top[x]],dfn[x],1,n,z,1);  //修改x到它的重链头的链
		x=pre[top[x]];
	}
	if(depth[x]>depth[y]){
		swap(x,y);
	}
	modify(dfn[x],dfn[y],1,n,z,1);
}
int qchain(int x,int y){
	int sum=0;
	while(top[x]!=top[y]){
		if(depth[top[x]]<depth[top[y]]){
			swap(x,y);
		}
		sum=(sum+query(dfn[top[x]],dfn[x],1,n,1))%mod;
		x=pre[top[x]];
	}
	if(depth[x]>depth[y]){
		swap(x,y);	
	}
	sum=(sum+query(dfn[x],dfn[y],1,n,1))%mod;
	return sum;
}
signed main(){
	cin>>n>>m>>r>>mod;
	for(int i=1;i<=n;i++){
		cin>>v[i];
	}
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		vec[u].push_back(v);
		vec[v].push_back(u);
	}
	dfs1(r,-1);
	dfs2(r,r);
	build(1,n,1);
	while(m--){
		int check;
		cin>>check;
		if(check==1){
			int x,y,z;
			cin>>x>>y>>z;
			mchain(x,y,z);
		}
		else if(check==2){
			int x,y;
			cin>>x>>y;
			cout<<qchain(x,y)<<"\n";
		}
		else if(check==3){
			int x,z;
			cin>>x>>z;
			mson(x,z);
		}
		else{
			int x;
			cin>>x;
			cout<<qson(x)<<"\n";
		}
	}
}