poj 3321 Apple Tree

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

题目大意：

给你N个点组成的树 每个点初始化为1 有两种操作

C x：改变点的值 是1变0 是0变1

Q x：问以x为根的子树上点值的和

思路：

主要是把树映射到树状数组中

一遍dfs把点重新编号 low和high low表示刚搜到此点时的计数，high是搜玩其子树内所有节点后回来的计数编号

比如说又n个点那么根节点1的low=1，high=2*n

这样就映射到树状数组中啦。

代码及其注释：

#include<iostream>

#include<cstdio>

#include<cstring>

#include<string>

#include<cmath>

#include<queue>

#include<algorithm>

#include<stack>



using namespace std;



const int N=100010;

struct node

{

    struct tt *next;

}mem[N];

struct tt

{

    struct tt *next;

    int j;

};

int n,I;//n is the number of apples I 是全局计数

int c[N*2];

int high[N];

int low[N];

bool visited[N];

int apple[N];//某个点是否有苹果

void build(int i,int j)//建树

{

    struct tt *t=new tt;

    t->j=j;

    t->next=mem[i].next;

    mem[i].next=t;

}

void dele()//一定的清理

{

    for(int i=1;i<=n;++i)

    mem[i].next=NULL;

}

int lowbit(int x)

{

    return x&(-x);

}

void Add(int x,int k)

{

    for(int i=x;i<=2*n;i+=lowbit(i))

    {

        c[i]+=k;

    }

}

int Sum(int x)

{

    int sum=0;

    for(int i=x;i>0;i-=lowbit(i))

    {

        sum+=c[i];

    }

    return sum;

}

void dfs(int x)//深搜 计数

{

    visited[x]=true;

    ++I;

    low[x]=I;

    struct tt *t=new tt;

    t=mem[x].next;

    while(t!=NULL)

    {

        if(visited[t->j]==false)

        {

            dfs(t->j);

        }

        t=t->next;

    }

    ++I;

    high[x]=I;



}

int main()

{

   int m;

   while(scanf("%d",&n)!=EOF)

   {

       memset(c,0,sizeof(c));

       for(int i=1;i<n;++i)

       {

           int l,r;

           scanf("%d %d",&l,&r);

           build(l,r);

           build(r,l);

       }

       I=0;

       memset(visited,false,sizeof(visited));

       dfs(1);

       for(int i=1;i<=n;++i)

       {

           apple[i]=1;

       }

       for(int i=1;i<=2*n;++i)

       {

           Add(i,1);//初始化树状数组

       }

       scanf("%d",&m);

       while(m--)

       {

           char c;

           int k;

           getchar();

           scanf("%c %d",&c,&k);

           if(c=='Q')

           {

               printf("%d\n",(Sum(high[k])-Sum(low[k]-1))/2);

           }else

           {

               if(apple[k]==1)

               {

                   apple[k]=0;

                   Add(low[k],-1);

                   Add(high[k],-1);

               }

               else

               {

                   apple[k]=1;

                   Add(low[k],1);

                   Add(high[k],1);

               }

           }

       }

       dele();

   }

   return 0;

}