【模板】树链剖分

题目描述
如题，已知一棵包含N个结点的树（连通且无环），每个节点上包含一个数值，需要支持以下操作：

操作1： 格式： 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z

操作2： 格式： 2 x y 表示求树从x到y结点最短路径上所有节点的值之和

操作3： 格式： 3 x z 表示将以x为根节点的子树内所有节点值都加上z

操作4： 格式： 4 x 表示求以x为根节点的子树内所有节点值之和

输入格式
第一行包含4个正整数N、M、R、P，分别表示树的结点个数、操作个数、根节点序号和取模数（即所有的输出结果均对此取模）。

接下来一行包含N个非负整数，分别依次表示各个节点上初始的数值。

接下来N-1行每行包含两个整数x、y，表示点x和点y之间连有一条边（保证无环且连通）

接下来M行每行包含若干个正整数，每行表示一个操作，格式如下：

操作1： 1 x y z

操作2： 2 x y

操作3： 3 x z

操作4： 4 x

输出格式
输出包含若干行，分别依次表示每个操作2或操作4所得的结果（对P取模）

输入输出样例
输入
5 5 2 24
7 3 7 8 0
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
输出
2
21
说明/提示

时空限制：1s，128M

（ 其实，纯随机生成的树LCA+暴力是能过的，可是，你觉得可能是纯随机的么233 ）

#include<cstdio>
#include<algorithm>
#define si 100010
#define re register int
#define ls(p) (p<<1)
#define rs(p) (p<<1|1)
using namespace std;
struct edge {
	int nex,to;
}e[si<<1];
int n,m,r,P,tot,head[si],w[si],wt[si],a[si<<2],tag[si<<2];
int cnt,son[si],id[si],fa[si],d[si],sz[si],top[si];
inline int read() {
	int x=0,cf=1;
	char ch=getchar();
	while(ch<'0'||ch>'9') {
		if(ch=='-') cf=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		x=(x<<3)+(x<<1)+ch-'0';
		ch=getchar();
	}
	return x*cf;
}
inline void add(int x,int y) {
	e[++tot].to=y,e[tot].nex=head[x],head[x]=tot;
}
inline void dfs1(int x,int f,int deep) {
	d[x]=deep,fa[x]=f,sz[x]=1;
	int maxson=-1;
	for(re i=head[x];i;i=e[i].nex) {
		int y=e[i].to;
		if(y==f) continue;
		dfs1(y,x,deep+1);
		sz[x]+=sz[y];
		if(sz[y]>maxson) son[x]=y,maxson=sz[y];
	}
}
inline void dfs2(int x,int topf) {
	id[x]=++cnt,wt[cnt]=w[x],top[x]=topf;
	if(!son[x]) return; dfs2(son[x],topf);
	for(re i=head[x];i;i=e[i].nex) {
		int y=e[i].to;
		if(y==fa[x]||y==son[x]) continue;
		dfs2(y,y);
	}
}
inline void pushup(int p) {
	a[p]=(a[ls(p)]+a[rs(p)])%P;
}
inline void pushdown(int p,int x) {
	if(!tag[p]) return;
	tag[ls(p)]+=tag[p];
	tag[rs(p)]+=tag[p];
	(a[ls(p)]+=tag[p]*(x-(x>>1)))%=P;
	(a[rs(p)]+=tag[p]*(x>>1))%=P;
	tag[p]=0; return;
}
inline void build(int p,int l,int r) {
	if(l==r) {
		a[p]=wt[l]%P;
		return;
	}
	int mid=(l+r)>>1;
	build(ls(p),l,mid);
	build(rs(p),mid+1,r);
	pushup(p); return;
}
inline void update(int p,int l,int r,int nl,int nr,int k) {
	if(nr<l||r<nl) return;
	if(nl<=l&&r<=nr) {
		tag[p]+=k;
		a[p]+=k*(r-l+1);
		return;
	}
	pushdown(p,r-l+1);
	int mid=(l+r)>>1;
	if(nl<=mid) update(ls(p),l,mid,nl,nr,k);
	if(nr>mid) update(rs(p),mid+1,r,nl,nr,k);
	pushup(p); return;
}
inline int query(int p,int l,int r,int nl,int nr) {
	int res=0;
	if(nl<=l&&r<=nr) return a[p]%P;
	pushdown(p,r-l+1);
	int mid=(l+r)>>1;
	if(nl<=mid) (res+=query(ls(p),l,mid,nl,nr))%=P;
	if(nr>mid) (res+=query(rs(p),mid+1,r,nl,nr))%=P;
	return res%P;
}
inline int qRange(int x,int y) {
	int ans=0;
	while(top[x]!=top[y]) {
		if(d[top[x]]<d[top[y]]) swap(x,y);
		(ans+=query(1,1,n,id[top[x]],id[x]))%=P;
		x=fa[top[x]];
	}
	if(d[x]>d[y]) swap(x,y);
	return (ans+query(1,1,n,id[x],id[y]))%P;
}
inline void updRange(int x,int y,int k) {
	while(top[x]!=top[y]) {
		if(d[top[x]]<d[top[y]]) swap(x,y);
		update(1,1,n,id[top[x]],id[x],k%P);
		x=fa[top[x]];
	}
	if(d[x]>d[y]) swap(x,y);
	update(1,1,n,id[x],id[y],k%P);
}
int main() {
	n=read(),m=read(),r=read(),P=read();
	for(re i=1;i<=n;i++) w[i]=read();
	for(re i=1;i<n;i++) {
		int x=read(),y=read();
		add(x,y),add(y,x);
	}
	dfs1(r,0,1),dfs2(r,r);
	build(1,1,n);
	while(m--) {
		int q=read(),x=read();
		if(q==1) {
			int y=read(),z=read();
			updRange(x,y,z%P);
		}
		else if(q==2) {
			int y=read();
			printf("%d\n",qRange(x,y));
		}
		else if(q==3) {
			int y=read();
			update(1,1,n,id[x],id[x]+sz[x]-1,y);
		}
		else printf("%d\n",query(1,1,n,id[x],id[x]+sz[x]-1));
	}
	return 0;
}