poj 3321

http://poj.org/problem?id=3321

题意：问某个节点的子节点以及它自己总共有多少个苹果，每C一次的话，则代表如果这个节点有苹果，则把苹果拿下，如果没有，则添加上去

思路：这个思路确实我没有想到过，用dfs从新建立一个新的序列，然后对这个新的序列求和（线段树和树状数组都可以）

关于DFS建立一个新的序列是这样的

就是从根节点开始dfs，用一个结构题记录这个节点是从哪个节点开始进入的（它本身的序列），以及最后的一个子节点是哪个节点

这样的话，就相当与构成了一个序列，就是从l->r就是它的编号，也正是因为这样，所以可以用树状数组来做

不理解可以把树构造出来，然后自己在列一下就会清楚

#include <stdio.h>
#include <string.h>
#include <iostream>
#define maxn 200008
using namespace std;

int n, pos, q,k;
int head[maxn];
int tree[maxn];

struct T {
    int l, r;
}num[maxn];

struct Node {
    int u, to;
}node[maxn];

void addedge(int u, int v)
{
    node[pos].u = v;
    node[pos].to = head[u];
    head[u] = pos++;
}

void add(int x, int num)
{
    while (x <= n)
    {
        tree[x] += num;
        x += x&-x;
    }
}

int read(int x)
{
    int sum = 0;
    while (x)
    {
        sum += tree[x];
        x -= x&-x;
    }
    return sum;
}

void dfs(int root)
{
    num[root].l = k;
    for (int i = head[root]; i != -1; i = node[i].to)
        k++,dfs(node[i].u);
    num[root].r = k;
}

int main()
{
   // freopen("in.txt","r",stdin);
    bool vis[maxn];
    int u, v;
    char tmp;
    k = 1;
    pos = 1;
    memset(head,-1,sizeof(head));
    memset(vis,false,sizeof(vis));
    scanf("%d", &n);
    for(int i = 1;i<n;i++)
    {
        scanf("%d%d", &u, &v);
        addedge(u, v);
    }
    dfs(1);
    scanf("%d",&u);
    for(int i = 1;i<=n;i++)
    {
        vis[i] = true;
        add(i,1);
    }
    getchar();
    while(u--)
    {
        scanf("%c%d",&tmp,&v);
        if(tmp=='Q')
        {
            printf("%d
",read(num[v].r)-read(num[v].l-1));
        }else {
            if(vis[v])
            {
                vis[v] = false;
                add(num[v].l,-1);
            }
            else
            {
                 vis[v] = true;
                 add(num[v].l,1);
            }

        }
        getchar();
    }
    return 0;
}