poj 3321 Apple Tree 树状数组 dfs序

题目链接

参考资料：

http://www.cnblogs.com/gj-Acit/p/3236843.html  ——再见~雨泉

题意：

给一棵树，初始时每个节点的 val 均为 1，现有一些操作，

Q：将某个节点的 val 取反

C：询问以某个节点为根的子树的 val 总和

思路：

dfs 一遍，将每个点的编号映射到 dfs 序，以 dfs 序来建树状数组。

每个点顺便记录两个值，le[ ] 为其自身的 dfs 序，ri[ ] 为其孩子中最大的 dfs 序。

那么查询某个节点为根的子树即为树上 le[ ] 到 ri[ ] 这连续的一段，直接用树状数组求两个前缀和作差即可。

关键的想法还是编号映射到 dfs 序。

Code:

#include <cstdio>#include <cstring>#include <algorithm>#define maxn 100010using namespace std;typedef long long LL;struct Edge {    int to, ne;    Edge(int a = 0, int b = 0) : to(a), ne(b) {}}edge[maxn * 2];int le[maxn], ri[maxn], ne[maxn], c[maxn], tot, cnt, n;bool flag[maxn];inline int max(int a, int b) { return a > b ? a : b; }void add(int u, int v) {    edge[tot] = Edge(v, ne[u]);    ne[u] = tot++;}void dfs(int u, int fa) {    le[u] = ++cnt;    int maxx = cnt;    for (int i = ne[u]; i != -1; i = edge[i].ne) {        Edge e = edge[i]; int v = e.to;        if (v == fa) continue;        dfs(v, u);        maxx = max(maxx, ri[v]);    }    ri[u] = maxx;}inline int lowbit(int x) { return x & (-x); }int query(int x) {    int ret = 0;    while (x) {        ret += c[x];        x -= lowbit(x);    }    return ret;}void modify(int x, int delta) {    while (x <= n) {        c[x] += delta;        x += lowbit(x);    }}void work() {    memset(c, 0, sizeof(c)); memset(ne, -1, sizeof(ne)); tot = cnt = 0;    for (int i = 1; i <= n; ++i) { modify(i, 1); flag[i] = 1; }    for (int i = 1; i < n; ++i) {        int x, y;        scanf("%d%d", &x, &y);        add(x, y); add(y, x);    }    dfs(1, -1);    int q;    scanf("%d", &q);//    for (int i = 1; i <= n; ++i) printf("%d %d\n", le[i], ri[i]);    while (q--) {        char ch; int x;        scanf("\n%c %d", &ch, &x);        if (ch == 'Q') {            printf("%d\n", query(ri[x]) - (le[x] == 1 ? 0 : query(le[x] - 1)));        }        else {            x = le[x];            if (flag[x]) modify(x, -1);            else modify(x, 1);            flag[x] ^= 1;        }    }}int main() {//    freopen("in.txt", "r", stdin);    while (scanf("%d", &n) != EOF) work();    return 0;}