poj-3321(dfs+树状数组）

http://acm.hust.edu.cn:8080/judge/problem/viewProblem.action?id=10486

思路：树状数组。这道题重点怎么建立树到树状数组的映射关系：利用dfs遍历树，对每个节点进行两次编号，第一次搜到第i个节点时的深度dep，为这个节点管辖区间的上限low[i](也为这个节点的下标)，然后搜这个节点的子节点，最后搜回来后的深度dep，为这个节点管辖区间的下限high[i]，如下图所示。接下来就是树状数组部分了。

[]中的第一个数，为树中这个点映射到线性数组中的所在位置。例如[1,5]中的1代表着映射之后，树的第一个结点在数组中也是第一个位置;

#include<iostream>
using namespace std;
#define max 100005
struct 
{
	int u,v,next;
}t[max];
int low[max],high[max],c[max],n,first[max],h;
int lowbit(int x)
{
	return x&(-x);
}
void updata(int i,int j)
{
	while(i<=n)
	{
		c[i]+=j;
		i+=lowbit(i);
	}
}
int getsum(int x)
{
	int sum=0;
	while(x>0)
	{
		sum+=c[x];
		x-=lowbit(x);
	}
	return sum;
}
void dfs(int u)                             //将树形结构映射成线性结构
{
	low[u]=++h;                             
	for(int i=first[u];i!=-1;i=t[i].next)        //first这个数组用来判断循环是否结束，当==-1则结束
	{
		dfs(t[i].v);                              //要是first数组!=-1，说明此处树有子节点，那么需要转换；i=t[i].next结点之后
	}                                          //（first结点的值==u或者-1的）所以要历遍玩first结点(t[i].u结点)后，还要历遍
	high[u]=h;                             //t[i].u后面的t[i].v结点
}
int main()
{
	while(scanf("%d",&n)>0)
	{
		memset(first,-1,sizeof(first));
		memset(c,0,sizeof(c));
		int i,k=0;
		for(i=0;i<n-1;i++)              //建树
		{
			int u,v;
			scanf("%d%d",&u,&v);
			k++;
			t[k].u=u;
			t[k].v=v;
			t[k].next=first[u];                 //让这一个结点连在下一个结点之后
			first[u]=k;                    //在dfs中，方便转换
		}
		for(i=1;i<=n;i++)             //初始值全部为1
			updata(i,1);
		h=0;
		dfs(1);
		int m;
		scanf("%d",&m);
		while(m--)
		{
			char s[5];
			int tmp;
			scanf("%s%d",s,&tmp);
			if(s[0]=='Q')
			{
				printf("%d\n",getsum(high[tmp])-getsum(low[tmp]-1));
			}
			else
			{
				int ans=getsum(low[tmp])-getsum(low[tmp]-1);      //在思路里面已经说了，low[tmp]代表的是树的第tmp个结点  
				if(ans==0)
					ans=1;
				else
					ans=-1;
				updata(low[tmp],ans);
			}
		}
	}
	return 0;
}