玄关求调,样例输出2个0
查看原帖
玄关求调,样例输出2个0
575480
mxdyzmx楼主2024/7/26 10:44

RT

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10;


vector<int> h[N];
int a[N], top[N], dep[N], fa[N], size[N], son[N], id[N], cnt = 0;
int mod, n, m, r;

class SegTree {
public:
#define ls(p) ((p) << 1)
#define rs(p) ((p) << 1 | 1)
#define up(p) (t[ls(p)] + t[rs(p)])
#define mid (((l) + (r)) >> 1)
	void build(int p, int l, int r) {
		lzy[p] = 0;
		if (l == r) {
			t[p] = a[l] % mod;
			return;
		}
		build(ls(p), l, mid);
		build(ls(p), mid + 1, r);
		t[p] = up(p);
	}
	inline void f(int p, int l, int r, int k) {
		lzy[p] = lzy[p] + k;
		t[p] = (t[p] + k * (r - l + 1)) % mod;
	}
	inline void push_down(int p, int l, int r) {
		f(ls(p), l, mid, lzy[p]);
		f(rs(p), mid + 1, r, lzy[p]);
		lzy[p] = 0;
	}
	inline void update(int p, int l, int r, int L, int R , int k) {
		if (L <= l && r <= R) f(p, l, r, k);
		else {
			push_down(p, l, r);
			if (L <= mid) update(ls(p), L, R, l, mid, k);
			if (R > mid) update(rs(p), L , R , mid + 1, r, k);
			t[p] = up(p);
		}
	}
	inline int query(int p, int l, int r, int L, int R) {
		int res = 0;
		if (L <= l || r <= R) {
			res += t[rs(p)];
			res %= mod;
		} else {
			push_down(p, l, r);
			if (L <= mid) res += query(ls(p), l, mid, L, R);
			if (R > mid) res += query(rs(p), mid + 1, r, L, R);
		}
		return res;
	}
private:
	int t[N << 2], lzy[N << 2];
} T;

void dfs1(int x, int f, int deep) {
	dep[x] = deep;
	fa[x] = f;
	size[x] = 1;
	int mson = -1;
	for (int i : h[x]) {
		if (i == f) continue;
		dfs1(i, x, deep + 1);
		size[x] += size[i];
		if (size[i] > mson) son[x] = i, mson = size[i];
	}
}

void dfs2(int x, int topf) {
	id[x] = ++cnt;
	a[cnt] = a[x];
	top[x] = topf;
	if (!son[x]) return;
	dfs2(son[x], topf);
	for (int i : h[x]) {
		if (i == fa[x] || i == son[x]) continue;
		dfs2(i, i);
	}
}

int qRange(int x, int y) {
	int ans = 0;
	while (top[x] != top[y]) {
		if (dep[top[x]] < dep[top[y]]) x ^= y, y ^= x, x ^= y;
		ans += T.query(1, 1, n, id[top[x]], id[x]);
		ans %= mod;
		x = fa[top[x]];
	}
	if (dep[x] > dep[y]) x ^= y, y ^= x, x ^= y;
	ans += T.query(1, 1, n, id[x], id[y]);
	return ans % mod;
}

int qSon(int x) {
	return T.query(1, 1, n, id[x], id[x] + size[x] - 1);
}

void updRange(int x, int y, int k) {
	k %= mod;
	while (top[x] != top[y]) {
		if (dep[top[x]] < dep[top[y]]) swap(x, y);
		T.update(1, 1, n, id[top[x]], id[x], k);
		x = fa[top[x]];
	}
	if (dep[x] > dep[y])swap(x, y);
	T.update(1, 1, n, id[x], id[y], k);
}

void updSon(int x, int k) {
	T.update(1, 1, n, id[x], id[x] + size[x] - 1, k);
}

int main() {
	scanf("%d%d%d%d",&n, &m, &r, &mod);
	for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
	for(int i = 1; i < n; i++) {
		int a, b;
		scanf("%d%d", &a, &b);
		h[a].push_back(b);
		h[b].push_back(a);
	}
	dfs1(r, 0, 1);
	dfs2(r, r);
	T.build(1, 1, n);
	while(m--) {
		int op,x,y,z;
		scanf("%d", &op);
		if(op == 1) {
			scanf("%d%d%d", &x, &y, &z);
			updRange(x, y, z);
		}else if(op == 2) {
			scanf("%d%d", &x, &y);
			printf("%d", qRange(x, y));
		}else if(op == 3) {
			scanf("%d%d", &x, &y);
			updSon(x,y);
		}else if(op == 4) {
			scanf("%d", &x);
			printf("%d", qSon(x));
		}
	}
	return 0;
}
2024/7/26 10:44
加载中...