POJ-3321 Apple Tree---树状数组+DFS序

题目链接：

https://vjudge.net/problem/POJ-3321

题目大意：

题目大意级是说，给你一颗树，最初每个节点上都有一个苹果，有两种操作：修改（即修改某一个节点，修改时这一个节点苹果从有到无，或从无到有）和查询（查询某一个节点他的子树上有多少个苹果）。

解题思路：

对树的每个节点进行编号，按照先序遍历进行编号，这样可以处理出一段区间，使得区间内都是点i的子树，求和的时候只要求区间和即可。

黑色是原来的编号，蓝色为用先序遍历得到的编号

超级坑的一点：

这里用vector<int>G[maxn]会超时

用vector<vector<int> > G(maxn)就过了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<map>
#include<stack>
#include<vector>
#include<list>
#include<deque>
#include<sstream>
#include<cctype>
#define REP(i, n) for(int i = 0; i < (n); i++)
#define FOR(i, s, t) for(int i = (s); i < (t); i++)
#define MEM(a, x) memset(a, x, sizeof(a));
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e5 + 10;
const double eps = 1e-10;
const int INF = 1 << 30;
const int dir[4][2] = {1,0,0,1,0,-1,-1,0};
const double pi = 3.1415926535898;
int T, n, m, cases, tot;
int tree[maxn], lef[maxn], rig[maxn], s[maxn];///s数组记录该节点是否有树
vector<vector<int> >a(maxn);
void dfs(int x)
///dfs给树节点编号，分别编成左节点和右节点，
///左节点就是本身，右节点是可到达的节点的最大编号，
///之后查询节点x的子树就可以直接查询[le[x], rig[x]]的区间和
{
    lef[x] = tot;
    for(int i = 0; i < a[x].size(); i++)
    {
        tot++;
        dfs(a[x][i]);
    }
    rig[x] = tot;
}
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret = 0;
    while(x > 0)
    {
        ret += tree[x];
        x -= lowbit(x);
    }
    return ret;
}
int add(int x, int d)
{
    while(x <= n)
    {
        tree[x] += d;
        x += lowbit(x);
    }
}
int main()
{
    int x, y;
    char c[5];
    while(scanf("%d", &n) != EOF)
    {
        MEM(lef, 0);
        MEM(rig, 0);
        MEM(s, 0);
        MEM(tree, 0);
        for(int i = 0; i < maxn; i++)a[i].clear();
        for(int i = 1; i < n; i++)
        {
            scanf("%d%d", &x, &y);
            a[x].push_back(y);
        }
        tot = 1;
        dfs(1);
        for(int i = 1; i <= n; i++)
        {
            s[i] = 1;
            add(i, 1);
        }
        scanf("%d", &m);
        while(m--)
        {
            scanf("%s%d", c, &x);
            if(c[0] == 'Q')
            {
                cout << sum(rig[x]) - sum(lef[x] - 1) << endl;
            }
            else if(c[0] == 'C')
            {
                if(s[x])add(lef[x], -1);
                else add(lef[x], 1);
                s[x] = !s[x];///更新该点是否有苹果
            }
        }
    }
    return 0;
}