poj 3321 树状数组

/*
题意：
     一棵具有n个节点的树，一开始，每个节点上都有一个苹果。
现在给出m组动态的操作：(C,i)是摘掉第i个节点上面的苹果(若苹果不存在，则为加上一个苹果)，
(Q,i)是查询以第i个节点为根的子树有几个苹果(包括第i个节点)。
解法：
    只是要注意把深搜的时间戳当作是数组的下标，而且如果是根节点，
    为保证唯一性，结束时间和开始时间应该是一个值，
    而非根节点的结束时间应为所有根节点结束时间的最晚的
*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=100002;
int NE,n,m,cnt,c[maxn],low[maxn],high[maxn],head[maxn],vis[maxn];
struct node
{
    int u,v,next;
} Edge[maxn];
void addEdge(int u,int v)
{
    Edge[NE].u=u,Edge[NE].v=v,Edge[NE].next=head[u];
    head[u]=NE++;
}
void dfs(int u,int fa)
{
    int i,j,v;
    cnt++;
    low[u]=cnt;
    for(i=head[u]; i!=-1; i=Edge[i].next)
    {
        v=Edge[i].v;
        if(v==fa)continue;
        dfs(v,u);
    }
    high[u]=cnt;
}
int lowbit(int i)
{
    return i&(-i);
}
void add(int i,int d)
{
    while(i<=n)
    {
        c[i]+=d;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int ret=0;
    while(i)
    {
        ret+=c[i];
        i-=lowbit(i);
    }
    return ret;
}
int main()
{
    int i,j,u,v,root;
    char Q[2];
    while(scanf("%d",&n)!=EOF)
    {
        cnt=NE=0;
        memset(head,-1,sizeof(head));
        memset(c,0,sizeof(c));
        memset(vis,0,sizeof(vis));
        memset(low,0,sizeof(low));
        memset(high,0,sizeof(high));
        for(i=1; i<n; i++)
        {
            scanf("%d%d",&u,&v);
            addEdge(u,v);
        }
        dfs(1,0);
        for(i=1; i<=n; i++) add(i,1);
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s%d",&Q,&root);
            if(Q[0]=='Q') printf("%d\n",sum(high[root])-sum(low[root]-1));
            else
            {
                if(!vis[low[root]])
                    add(low[root],-1);
                else add(low[root],1);
                vis[low[root]]^=1;
            }
        }
    }
    return 0;
}
/*
3
1 2
1 3
3
Q 1
C 2
Q 1
*/