Luogu P3384(树链剖分，裸题)

一个代码模板。

题目描述

如题，已知一棵包含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取模）

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

2
21

#include <bits/stdc++.h>
#define mem(x,v) memset(x,v,sizeof(x)) 
#define go(i,a,b)  for (int i = a; i <= b; i++)
#define og(i,a,b)  for (int i = a; i >= b; i--)
#define MID(a,b) (a + b) / 2
#define lson now << 1
#define rson now << 1 | 1
using namespace std;
typedef long long LL;
const double EPS = 1e-10;
const int INF = 0x3f3f3f3f;
const int N = 1e5+100;
int n,m,r,p; //正常读入。
int lazy[N<<2],val[N<<2]; //线段树。
int cnt = -1,Head[N],Next[N],to[N],w[N],wt[N];//权值，邻接表
int son[N],id[N],f[N],tim,size[N],top[N],dep[N];//树剖数组。
// 重儿子，入队时间戳，父亲节点，子树大小，链的顶端，深度。
int res = 0;
void Add_edge(int u, int v){
	to[++cnt] = v;
	Next[cnt] = Head[u];
	Head[u] = cnt;
}
void dfs1(int u, int fa, int deep){ //解决深度，子树大小，重儿子，父亲节点。
	dep[u] = deep; //deep
	f[u] = fa; size[u] = 1;// father , size of node.
	for (int i = Head[u]; i != -1; i = Next[i]){
		int v = to[i];
		if (v == fa) continue;
		dfs1(v,u,deep+1);
		size[u] += size[v]; //size of u.
		if (size[v] > size[son[u]]) son[u] = v; // 找u 的重儿子。
	}
}
void dfs2(int u, int topf){
	id[u] = ++tim; //时间戳
	wt[tim] = w[u];//权值
	top[u] = topf; //找链顶。
	if (!son[u]) return;
	dfs2(son[u],topf); //先找重儿子，
	for (int i = Head[u]; i != -1; i = Next[i]){ //再找轻儿子。
		int v = to[i];
		if (v == f[u] || v == son[u]) continue; //如果是重儿子，或者父亲节点，跳过。
		dfs2(v,v);
	}
}
// 线段树操作
void push_down(int now, int len){
	if (lazy[now] == 0) return;
	val[lson] = (val[lson] + lazy[now]*(len >> 1)) % p;
	val[rson] = (val[rson] + lazy[now]*(len - (len >> 1))) % p;
	lazy[lson] = (lazy[lson] + lazy[now]);
	lazy[rson] = (lazy[rson] + lazy[now]);
	lazy[now] = 0;
}
void Build(int now, int a, int b){
	if (a + 1 == b){
		val[now] = wt[a] % p;
		return;
	}
	int m = MID(a,b);
	Build(lson,a,m); Build(rson,m,b);
	val[now] = (val[lson] + val[rson]) % p;
}
int Query(int now, int a, int b, int l, int r){
	int ans = 0;
	if (l <= a && r >= b - 1){
		return (val[now] % p);
	}
	int m = MID(a,b);
	push_down(now,b-a);
	if (l < m) ans = (ans + Query(lson,a,m,l,r))%p;
	if (r >= m) ans = (ans + Query(rson,m,b,l,r)) % p;
	return ans;
}

void Insert(int now, int a, int b, int l, int r, int k){
	if (l <= a && r >= b - 1){
		val[now] = (val[now] + k * (b - a))%p;
		lazy[now] += k;
		return;
	}
	push_down(now,b-a);
	int m = MID(a,b);
	if (l < m) Insert(lson,a,m,l,r,k);
	if (r >= m) Insert(rson,m,b,l,r,k);
	val[now] = (val[lson] + val[rson]) % p;
}
// 线段树操作
int qRange(int x, int y){
	int ans = 0;
	while(top[x] != top[y]){
		if (dep[top[x]] < dep[top[y]]) swap(x,y); // 把 x 变成深度大的点。
		ans = (ans + Query(1,1,n+1,id[top[x]],id[x])) % p;//处理路径上的和。
		x = f[top[x]]; //到达链顶的父亲节点，就是换一条链。
	}
	if (dep[x] < dep[y]) swap(x,y);//在同一条链，不一定相等。把x 变成深度大的点。
	ans = (ans + Query(1,1,1+n,id[y],id[x])) % p; //处理路径上的和。
	return ans;
}
void upRange(int x, int y, int k){ //同上。
	k %= p;
	while(top[x] != top[y]){
		if (dep[top[x]] < dep[top[y]]) swap(x,y);
		Insert(1,1,n+1,id[top[x]],id[x],k);
		x = f[top[x]];
	}
	if (dep[x] < dep[y]) swap(x,y);
	Insert(1,1,n+1,id[y],id[x],k);
}
int qSon(int x){
	return Query(1,1,n+1,id[x],id[x]+size[x]-1);
}

void upSon(int x, int k){
	Insert(1,1,n+1,id[x],id[x] + size[x] -1,k);
}

int main(){
	scanf("%d%d%d%d",&n,&m,&r,&p);
	go(i,1,n) scanf("%d",&w[i]);
	mem(Head,-1);
	go(i,2,n){
		int x,y;
		scanf("%d%d",&x,&y);
		Add_edge(x,y); Add_edge(y,x);
	}
	dfs1(r,0,1);
	dfs2(r,r);
	Build(1,1,n+1);
	while(m--){
		int op,x,y,z;
		scanf("%d",&op);
		if (op == 1){
			scanf("%d%d%d",&x,&y,&z);
			upRange(x,y,z);
		} else 
		if (op == 2) {
			scanf("%d%d",&x,&y);
			printf("%d\n",qRange(x,y));
		} else
		if (op == 3){
			scanf("%d%d",&x,&y);
			upSon(x,y);
		} else {
			scanf("%d",&x);
			printf("%d\n",qSon(x));
		}
	}
	return 0;
}