POJ 3321 Apple Tree

题目链接：http://poj.org/problem?id=3321

刷POJ分类的时候遇到的题，只知道是树状数组，但是分支编号不连续，不晓得怎么用树状数组求和。

后来看了解题报告才知道需要把分支通过DFS映射成编号连续的，记录每个子树的起始位置和终止位置，然后再用树状数组求和。

即使明白了思路，代码实现上也还是有很多地方不理解。唉，数据结构学得实在有够糟糕……

树的邻接表表示法每次往表头插入节点也让我费解了好半天Orz……

#include <cstdio>
#include <cstring>
#include <cstdlib>

const int MAXN = 100005;

struct node    //邻接表保存树
{
    int v;
    int next;
};

struct SubTree  //每棵子树的起始点和终止点
{
    int st;
    int ed;
};

int N, Q;
int EdgeN, TimeFlag;
node D[MAXN];
SubTree R[MAXN];
int C[MAXN];       //树状数组求和用
int apple[MAXN];   //以u为根的子树上是否有苹果
int head[MAXN];    //以u为根的子树编号为head[u]

void AddEdge( int u, int v )   //向邻接表中添加一条边
{
    D[EdgeN].v = v;
    D[EdgeN].next = head[u];
    head[u] = EdgeN++;
    return;
}

void DFS( int cur )   //遍历树，给每个分支编号
{
    R[cur].st = TimeFlag;     //子树的起始位置
    for ( int i = head[cur]; i != -1; i = D[i].next )
        DFS( D[i].v );
    R[cur].ed = TimeFlag++;   //子树的终止位置
    return;
}

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

void Add( int x, int d )  //树状数组更新
{
    while ( x <= N )
    {
        C[x] += d;
        x += lowbit(x);
    }
    return;
}

int Sum( int x )    //求和
{
    int ret = 0;
    while ( x > 0 )
    {
        ret += C[x];
        x -= lowbit(x);
    }
    return ret;
}

int main()
{
    while ( ~scanf( "%d", &N ) )
    {
        EdgeN = 1;
        memset( head, -1, sizeof(head) );
        for ( int i = 1; i < N; ++i )
        {
            int u, v;
            scanf( "%d%d", &u, &v );
            AddEdge( u, v );
        }

        TimeFlag = 1;
        DFS( 1 );

        for ( int i = 1; i <= N; ++i )   //初始时整棵树上都有苹果
        {
            C[i] = lowbit(i);
            apple[i] = 1;
        }

        scanf( "%d", &Q );
        while ( Q-- )
        {
            char op[4];
            int u;
            scanf( "%s%d", op, &u );
            if ( op[0] == 'C' )
            {
                int val = 1;
                if ( apple[u] ) val = -1;
                apple[u] ^= 1;
                Add( R[u].ed, val );
            }
            else
                printf("%d\n", Sum( R[u].ed ) - Sum( R[u].st - 1 ) );
        }
    }
    return 0;
}