【树的遍历 树状数组】POJ_3321 Apple Tree

题意

给出一个苹果树的一些分叉(x,y)，表示y是x的子节点，给出m次操作，有两种：
1、C x 如果叉x有苹果就摘，没有就长，相当于取反。
2、Q x 查询叉x（包括x）的子节点上一共有多少苹果。

思路

先进行深度优先遍历求出这棵树的先序后序遍历顺序，然后我们进行C操作时就判断然后取反，进行Q操作时就用$find(end[x])-find(begin[x]),begin$是先序遍历，$end$是后序遍历，这样子我们可以求出$x$的子节点的苹果数量了。

代码

#include<cstdio>
struct node{
	int next,to;
}edge[100001];
int tot,n,m,x,tree[100001],head[100001],begin[100001],end[100001],apple[100001];
int c;
int lowbit(int x) {return x&-x;}
void add(int x,int v)
{
	for (;x<=n;x+=lowbit(x))
		tree[x]+=v;
}
int find(int x)
{
	int o=0;
	for (;x>0;x-=lowbit(x))
		o+=tree[x];
	return o;
}
void dfs(int x)
{
	begin[x]=tot;
	for (int i=head[x];i;i=edge[i].next) dfs(edge[i].to);
	end[x]=++tot;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&edge[i].to);//邻接表优化遍历
		edge[i].next=head[x];
		head[x]=i;
		add(i,1);//边输入边初始化
	}
	add(n,1);
	dfs(1);//求先序后序遍历
	scanf("%d",&m);
	for (int i=1;i<=m;i++)
	{
		scanf("\n%c %d",&c,&x);
		if (c=='C')
		{
			apple[x]^=1;//取反
			if (apple[x]) add(end[x],-1);//因为有苹果所以摘
			else add(end[x],1);//相反
		}
		else printf("%d\n",find(end[x])-find(begin[x]));
	}
}