POJ_3321_APPLE_TREE

poj上面的一道求子树上苹果的题目，网上看了很多题解，下面我来回忆一下，基本来源于大神的微博，http://blog.csdn.net/zhang20072844，我来做个搬运工。
先将树的n条边上节点重新标号，让每一棵子树上的节点编号构成一个连续的区间，然后利用dfs将第i个节点表示的区间下限，上限存进数组low[u],high[u]。树状数组c[i]=a[i-2^k+1]+a[i-2^k+2]+.....a[i]，也就是i-2^k+1到i之间的苹果总数，每一个a[i]表示树上一个分叉，i是经过重新标号后的表示节点的值。可以通过add（low[i],1）为第i（原苹果树中标号）个节点增加一个苹果，同时更新树状数组中和i（也就是元素a[low[i]]有关的c[j]值，这样查询low[i]到high[i]区间范围内的苹果数目便可以通过sum（high[i])-sum(low[i]-1）,其中sum（t）表示a[i]+a[2].....a[t]之和也就是这些节点上苹果数目。

 #include<stdio.h>
 #include<string.h>
 #define N 100010

 typedef struct
 {
     int to,next;
 } Edge;
 Edge edge[N];
 int low[N],high[N],vis[N],c[N*],head[N],pick[N];
 int dep,p,n;

 void addedge(int cu,int cv)
 {
     edge[p].to=cv;
     edge[p].next=head[cu];
     head[cu]=p++;
 }

 void dfs(int u)
 {
     low[u]=++dep;
     vis[u]=;
     int i;
     for(i=head[u]; i!=-; i=edge[i].next)
     {
         if(!vis[edge[i].to])
             dfs(edge[i].to);
     }
     high[u]=dep;
 }

 int lowbit(int x)
 {
     return x&(-x);
 }
 void add(int t,int v)
 {
     while(t<=n)
     {
         c[t]+=v;
         t+=lowbit(t);
     }
 }

 int sum(int t)
 {
     int res=;
     while(t>)
     {
         res+=c[t];
         t-=lowbit(t);
     }
     return res;
 }

 int main(void)
 {
     int m,i,u,v;
     memset(head,-,sizeof(head));
     scanf("%d",&n);
     for(i=; i<n; i++)
     {
         scanf("%d%d",&u,&v);
         addedge(u,v);
     }
     dfs();
     for(i=; i<=n; i++)
     {
         add(i,);
         pick[i]=;
     }
     scanf("%d",&m);
     while(m--)
     {
         getchar();
         char op;
         scanf("%c%d",&op,&i);
         if(op=='Q')
             printf("%d\n",sum(high[i])-sum(low[i]-));
         else
         {
             if(pick[i])
             {
                 add(low[i],-);
                 pick[i]=;
             }
             else
             {
                 add(low[i],);
                 pick[i]=;
             }
         }
     }
     return ;
 }