BZOJ 4034 树上操作

BZOJ 4034 树上操作


有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个
操作,分为三种:

操作 1 :把某个节点 x 的点权增加 a 。

操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 :询问某个节点 x 到根的路径中所有点的点权和。


做法 : 欧拉序列 + 线段树

先在树上做一遍 DFS, 按照树的欧拉序列建线段树,然后操作1就变成单点修改,操作2,就变成了区间内某些点+a,某些点-a.

那么如何确认区间内+a 和 -a 的数字的个数呢。 这里要用到一个flag数组来记录一下。 注意到我们的查询只关心欧拉序列上一段区间上的和,因此对于一段序列的修改,我们只要维护好区间和的大小即可。 于是区间内+a和-a的点可以相互抵消,我们只需要在区间上面加上a * (加a的位置数量-减a的位置数量) 即可。。。。

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

const int maxn = 200000 + 55;
const int N = maxn * 6;

int n, m, cnt = 0;
int a[maxn], b[N], dfn[maxn], low[maxn]; bool io[N];
vector<int> G[maxn];

inline void dfs(int x, int fat) {
	++ cnt; b[cnt] = a[x]; dfn[x] = cnt; io[cnt] = 1;
	for (int i = 0; i < G[x].size(); ++ i) {
		int v = G[x][i]; if (v == fat) continue;
		dfs(v, x);
	}
	++ cnt; b[cnt] = -a[x]; low[x] = cnt; io[cnt] = 0;
}

const int maxm = 2400010;

int L[maxm + 1], R[maxm + 1];
long long lazy[maxm + 1], flag[maxm + 1];
long long w[maxm + 1];

inline void build(int pos, int l, int r) {
	L[pos] = l; R[pos] = r; w[pos] = lazy[pos] = 0;
	if (l == r) {
		w[pos] = (long long) b[l];
		if (io[l]) flag[pos] = 1;
		else flag[pos] = -1;
		return;
	}
	int mid = (l + r) >> 1;
	build(pos << 1, l, mid);
	build(pos << 1 | 1, mid + 1, r);
	w[pos] = w[pos << 1] + w[pos << 1 | 1];
	flag[pos] = flag[pos << 1] + flag[pos << 1 | 1];
}

inline void pushdown(int x) {
	if (!lazy[x]) return;
	w[x << 1] += lazy[x] * flag[x << 1];
	w[x << 1 | 1] += lazy[x] * flag[x << 1 | 1];
	lazy[x << 1] += lazy[x];
	lazy[x << 1 | 1] += lazy[x];
	lazy[x] = 0;
}	

inline void update(int x, int l, int r, long long delta) {
	if (l <= L[x] && R[x] <= r) {
		w[x] = w[x] + flag[x] * delta;
		lazy[x] += delta;
		return;
	}
	pushdown(x);
	int mid = (L[x] + R[x]) >> 1;
	if (l > mid) update(x << 1 | 1, l, r, delta);
	else if (r <= mid) update(x << 1, l, r, delta);
	else {
		update(x << 1, l, mid, delta);
		update(x << 1 | 1, mid + 1, r, delta);
	}
w[x] = w[x << 1] + w[x << 1 | 1];
}	

inline long long query(int x, int l, int r) {
	if (l <= L[x] && R[x] <= r) {
		return w[x];
	}
	pushdown(x);
	int mid = (L[x] + R[x]) >> 1;
	if (l > mid) return query(x << 1 | 1, l, r);
	else if (r <= mid) return query(x << 1, l, r);
	else {
		return query(x << 1, l, mid) + query(x << 1 | 1, mid + 1, r);
	}
}

int main(void) {
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; ++ i) scanf("%d", &a[i]);
	for (int i = 1; i < n; ++ i) {
		int x, y; scanf("%d%d", &x, &y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	dfs(1, 0);
	build(1, 1, cnt);
	while (m --) {
		int op; scanf("%d", &op);
		if (op == 1) {
			int x, del; scanf("%d%d", &x, &del);
			update(1, dfn[x], dfn[x], (long long)del);
			update(1, low[x], low[x], (long long)del);
		} else if (op == 2) {
			int x, del; scanf("%d%d", &x, &del);
			update(1, dfn[x], low[x], (long long)del);
		} else if (op == 3) {
			int x; scanf("%d", &x);
			printf("%lld\n", query(1, 1, dfn[x]));
		}
	}
	return 0;
}

以上就是代码!是不是很精简啊!

posted @ 2018-07-14 11:49  Iamhx  阅读(116)  评论(0编辑  收藏  举报