Poj 3321 Apple Tree

题目大意：给定一棵树（题目中默认节点 1 为根），树的节点上可能有苹果，也可能没有。需要查询某个节点及其子节点上的苹果数量之和，并且支持节点上长苹果或苹果摘（将节点的值改为1或者改为0）。

思路：要用到树状数组来统计某个范围内元素的和，可是根据当前节点的编号，是没办法使用树状数组的。对于节点 i，我们必须让节点i的所有子节点的编号和 i 连续。对此我们需要做一些预处理。先建树，用DFS对树进行后序遍历，根据访问的先后顺序给节点重新编号。这样对于节点i，其所有子节点的编号都小于i，并且是连续的。当然并不是所有编号比i小的节点都是i的子节点，所以对于节点 i 我们必须记录它子节点当中的最小编号。

#include <iostream>
#include <stdio.h>
#include <memory.h>
using namespace std;
const int maxn=100010;
int c[maxn];
int n;// 元素的个数,表示1到n
int m;
struct node {
	int data;
	struct node *next;
};
struct node graph[maxn];
int sum1;
int end[maxn];
int v[maxn];
void DFS(int s) {
	
	v[s]=++sum1;
	struct node *p=graph[s].next;
	while (p) {
		if (v[p->data]==0)
			DFS(p->data);
		p=p->next;
	}
	end[s]=sum1;
}
//计算lowbit
int lowbit(int x) {
	return x&(-x);
}
//设原始矩阵为a,将a[i]加上val时对c所做的修改
void add(int i, int val) {
	while (i<=n) {
		c[i]+=val;
		i+=lowbit(i);
	}
} 
//求前i项元素的和
int get_sum(int i) {
	int sum=0;
	while (i>0) {
		sum+=c[i];
		i-=lowbit(i);
	}
	return sum;
}
int main()
{
	int i,j,s,t;
	
	char ch;

	scanf("%d",&n);
	for (i=1;i<=n-1;i++) {
		scanf("%d%d",&s,&t);
		struct node *p=new struct node;
		struct node *q=new struct node;
		p->data=t;
		p->next=graph[s].next;
		graph[s].next=p;

		q->data=s;
		q->next=graph[t].next;
		graph[t].next=q;
	}
	memset(v,0,sizeof(v));
	DFS(1);
	memset(c,0,sizeof(c));
	for (i=1;i<=n;i++)
		add(i,1);
	scanf("%d",&m);
	for (i=1;i<=m;i++) {
		getchar();
		scanf("%c%d",&ch,&j);
		if (ch=='C') {
			s=get_sum(v[j]);
			s=s-get_sum(v[j]-1);
			if (s==1) 
				add(v[j],-1);
			else
				add(v[j],1);
		}
		else {
			s=get_sum(end[j]);
			s=s-get_sum(v[j]-1);
			printf("%d\n",s);
		}
	}
	return 0;
}