POJ3321Apple Tree【dfs 树状数组】

题目大意：一棵树（不一定是二叉树！！），树的节点上本来都有一个苹果，要求完成以下操作：

1.指定某个节点，如果这个节点原本有苹果则拿去，如果没有苹果则填上一个苹果

2.询问某个节点以及其子树一共有多少个苹果

思路：dfs这棵树，记录下第一次到达这个节点的时间以及遍历离开的时间，于是一个节点就成了一个区间，左端点和右端点分别是开始遍历的时间和结束遍历的时间，区间里夹着的就是这个节点的子树！！区间端点记录有没有苹果，这样问题就变成了询问一个区间和的问题，而加减苹果就是对区间端点+1和-1 这两个操作都能用树状数组解决

 

#include<cstdio>

#include<string.h>

#include<iostream>

#define maxn 200009

using namespace std;

intnow=0,tree[maxn]={0},n,next[maxn]={0},root[maxn]={0},edge[maxn]={0},start[maxn]={0},end[maxn]={0};

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

 

void add(int x,int c)

{

   while (x<=2*n)

    {

       tree[x]+=c;

       x+=lowbit(x);

    }

}

int get(int x)

{

   int sum=0;

   for(int i=x;i>=1;i=i-lowbit(i))sum+=tree[i];

   return sum;

}

void addedge(int x,int y)

{

   next[++now]=root[x];

   edge[now]=y;

   root[x]=now;

}

void dfs(int j)

{

   start[j]=++now;

   for(int i=root[j];i!=0;i=next[i])dfs(edge[i]);

   end[j]=++now;

}

int main()

{

   int x,y,m,ope,flag;

   char ch[2];

   scanf("%d",&n);

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

    {

       scanf("%d%d",&x,&y);

       addedge(x,y);

    }

   now=0;

   dfs(1);

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

   add(i,1);

   scanf("%d",&m);

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

    {

       scanf("%s%d",ch,&ope);

       if (ch[0]=='Q')

        {

           printf("%d\n",(get(end[ope])-get(start[ope]-1))>>1);

       }

       else

       {

           flag=get(end[ope])-get(end[ope]-1);

           if (flag==1)flag=-1;else flag=1;

           add(start[ope],flag);

           add(end[ope],flag);

       }

    }

   return 0;

}

 

调试结果：1次AC