poj 3321 dfs + 树状数组

题意：

给一棵树，然后每个叶子节点上开始的时候有一棵苹果，然后有如下操作：

C x ， 如果 x 节点上有苹果，就摘下来，如果没有苹果，就长一个；

Q x，询问 x 这个父亲节点下共有几个苹果。

解析：

把树的深度遍历和树状数组组合的绝妙至极。

首先，考虑到怎么将一棵二维的树，拉成一维的到树状数组里面去维护个数。

dfs + 时间戳是一个非常妙的用法，刚好把s时间戳作为树状数组的左区间端点，e时间戳作为右区间端点。

这样处理就刚好把一个二维的树抽象成了一维，因为在当前这个父亲节点下的子节点的时间戳必然是在 （s，e）这个区间内的。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <climits>
#include <cassert>
#define LL long long
#define lson lo, mi, rt << 1
#define rson mi + 1, hi, rt << 1 | 1

using namespace std;
const int maxn = 100000 + 10;
const int inf = 0x3f3f3f3f;
const double eps = 1e-8;
const double pi = acos(-1.0);
const double ee = exp(1.0);

int n;
int c[maxn];

int lo[maxn], hi[maxn];
int mark[maxn];
int head[maxn];
int edgeNum;
int dep;

void init()
{
    memset(c, 0, sizeof(c));
    memset(head, -1, sizeof(head));
    memset(lo, -1, sizeof(lo));
    memset(hi, -1, sizeof(hi));
    edgeNum = 0;
    dep = 0;
}

struct Tree
{
    int to, next;
} edge[maxn];

void addEdge(int fr, int to)
{
    edge[edgeNum].to = to;
    edge[edgeNum].next = head[fr];
    head[fr] = edgeNum++;
}

void dfs(int u)
{
    lo[u] = ++dep;///debug
    for (int i = head[u]; i != -1; i = edge[i].next)
    {
        if (lo[edge[i].to] == -1)
            dfs(edge[i].to);
    }
    hi[u] = dep;
}

int lowbit(int x)
{
    return x & -x;
}

void update(int x, int num)
{
    while (x <= n)
    {
        c[x] += num;
        x += lowbit(x);
    }
}

int query(int x)
{
    int res = 0;
    while (x)
    {
        res += c[x];
        x -= lowbit(x);
    }
    return res;
}

int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
#endif // LOCAL
    while (~scanf("%d", &n))
    {
        init();
        for (int i = 1; i <= n; i++)
        {
            mark[i] = 1;
            update(i, 1);
        }
        for (int i = 1; i < n; i++)
        {
            int fr, to;
            scanf("%d%d", &fr, &to);
            addEdge(fr, to);
        }
        dfs(1);
//        for (int i = 1; i <= n; i++)
//            cout << lo[i] << " " << hi[i] << endl;
        int m;
        scanf("%d", &m);
        while (m--)
        {
            char op[3];
            scanf("%s", op);
            int rt;
            scanf("%d", &rt);
            if (op[0] == 'Q')
            {
                printf("%d\n", query(hi[rt]) - query(lo[rt] - 1));
            }
            if (op[0] == 'C')
            {
                if (mark[rt])
                {
                    update(lo[rt], -1);
                    mark[rt] = 0;
                }
                else
                {
                    update(lo[rt], 1);
                    mark[rt] = 1;
                }
            }

        }
    }
    return 0;
}