poj3321Apple Tree（树状数组）

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

刚一看题以为要建一颗树 看了下讨论说dfs

这里dfs遍历时设的标号很好 一个low一个high 包含了以这一节点为根节点的子树结点的所有标号

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<stdlib.h>
 using namespace std;
 #define N 100010
 #define lowbit(x) (x&(-x))
 struct node
 {
     int u,v,next;
 }ed[N<<];
 int head[N],t,low[N],g,high[N],vis[N],re[N],f[N];
 void init()
 {
     t = ;
     memset(head,-,sizeof(head));
 }
 void add(int u,int v)
 {
     ed[t].u = u;
     ed[t].v = v;
     ed[t].next = head[u];
     head[u] = t;
     t++;
 }
 void dfs(int u)
 {
     low[u] = ++g;
     vis[u] = ;
     int i;
     for(i = head[u] ; i != - ; i = ed[i].next)
     {
         int v = ed[i].v;
         if(!vis[v])
             dfs(v);
     }
     high[u] = g;
 }
 void add1(int i,int da)
 {
     while(i<=g)
     {
         re[i]+=da;
         i+=lowbit(i);
     }
 }
 int getsum(int i)
 {
     int sum = ;
     while(i)
     {
         sum+=re[i];
         i-=lowbit(i);
     }
     return sum;
 }
 int main()
 {
     int i,j,k,n,m,a,b;
     char s[];
     init();
     scanf("%d",&n);
     for(i = ; i < n ; i++)
     {
         scanf("%d%d",&a,&b);
         add(a,b);
         add(b,a);
     }
     dfs();
     for(i =  ; i <= g ; i++)
     {
         f[i] = -;
         add1(i,);
     }
     scanf("%d",&m);
     while(m--)
     {
         scanf("%s %d",s,&k);
         if(s[]=='Q')
         {
             printf("%d\n",getsum(high[k])-getsum(low[k]-));
         }
         else
         {
             add1(low[k],f[k]);
             f[k] = f[k]*-;
         }
     }
     return ;
 }