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;
}
以上就是代码!是不是很精简啊!