BZOJ 4034 树上操作【线段树】【dfs序】

4034: [HAOI2015]树上操作

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 6904 Solved: 2323
[ Submit][ Status][ Discuss]

Description

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个

操作，分为三种：

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

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

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

Input

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1

行每行三个正整数 fr, to ，表示该树中存在一条边 (fr, to) 。再接下来 M 行，每行分别表示一次操作。其中

第一个数表示该操作的种类（ 1-3 ），之后接这个操作的参数（ x 或者 x a ）。

Output

对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

Sample Input

5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

Sample Output

6
9
13

HINT

对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lson rt << 1, l, mid
#define rson rt << 1 | 1, mid + 1, r
#define mms(x, y) memset(x, y, sizeof x)
const int MAX = 2e6 + 7;
int num, cnt;
int head[MAX];
int flag[MAX], S[MAX];
int in[MAX], out[MAX];
ll a[MAX], tree[MAX], lazy[MAX], DFS[MAX];
struct node
{
    int v, next;
} e[MAX << 1];
void init()
{
    mms(DFS, 0);
    mms(lazy, 0);
    mms(flag, 0);
    mms(tree, 0);
    mms(head, -1);
    mms(S, 0);
    cnt = num = 1;
}
void add(int u, int v)
{
    e[num] = {v, head[u]};
    head[u] = num++;
}
void dfs(int u, int fa)
{
    DFS[in[u] = ++cnt] = a[u];
    S[cnt] = 1;
    for(int i = head[u]; i != -1; i = e[i].next)
        if(e[i].v != fa)
            dfs(e[i].v, u);
    DFS[out[u] = ++cnt] = -a[u];
    S[cnt] = -1;
}
void pushup(int rt)
{
    tree[rt] = tree[rt << 1] + tree[rt << 1 | 1];
    flag[rt] = flag[rt << 1] + flag[rt << 1 | 1];
}
void pushdown(int rt)
{
    if(lazy[rt]){
        tree[rt << 1] += flag[rt << 1] * lazy[rt];
        tree[rt << 1 | 1] += flag[rt << 1 | 1] * lazy[rt];
        lazy[rt << 1] += lazy[rt];
        lazy[rt << 1 | 1] += lazy[rt];
        lazy[rt] = 0;
    }
}
void build(int rt, int l, int r)
{
    if(l == r){
        tree[rt] = DFS[l];
        if(S[l] > 0)
            flag[rt] = 1;
        else
            flag[rt] = -1;
        return;
    }
    int mid = (l + r) >> 1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int rt, int l, int r, int x, int y, int v)
{
    if(x <= l && r <= y){
        tree[rt] += (ll)(flag[rt] * (ll)v);
        lazy[rt] += (ll)v;
        return;
    }
    int mid = (l + r) >> 1;
    pushdown(rt);
    if(x <= mid)
        update(lson, x, y, v);
    if(mid < y)
        update(rson, x, y, v);
    pushup(rt);
}
ll query(int rt, int l, int r, int x, int y)
{
    if(x <= l && r <= y)
        return tree[rt];
    int mid = (l + r) >> 1;
    ll ans = 0;
    pushdown(rt);
    if(x <= mid)
        ans += query(lson, x, y);
    if(mid < y)
        ans += query(rson, x, y);
    return ans;
}
int main()
{
    int n, m;
    init();
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++)
        scanf("%lld", &a[i]);
    for(int i = 1; i < n; i++){
        int u, v;
        scanf("%d%d", &u, &v);
        add(u, v);
        add(v, u);
    }
    dfs(1, 0);
    build(1, 1, 2 * n);
    while(m--){
        int s, x, y;
        scanf("%d", &s);
        if(s == 1){
            scanf("%d%d", &x, &y);
            update(1, 1, 2 * n, in[x], in[x], y);
            update(1, 1, 2 * n, out[x], out[x], y);
        }
        else if(s == 2){
            scanf("%d%d", &x, &y);
            update(1, 1, 2 * n, in[x], out[x], y);
        }
        else{
            scanf("%d", &x);
            ll ans = query(1, 1, 2 * n, 1, in[x]);
            printf("%lld\n", ans);
        }
    }
    return 0;
}