树状数组相关应用之多叉树子树问题

Poj-3321：Apple Tree


题意：
卡卡的房子外面有一棵苹果树。每年秋天，树上都会长出许多苹果。卡卡非常喜欢苹果，所以他一直在精心培育着这棵大苹果树。
这棵树有N个分叉，这些分叉由树枝相连。卡卡把叉的编号从1到N，根编号总是1。苹果会在叉子上生长，两个苹果不会在同一个叉子上生长。卡卡想知道一个子树上有多少个苹果，因为他研究的是苹果树的生产能力。
问题是一个新的苹果可能会在空叉子上生长，卡卡可能会从树上摘一个苹果作为甜点。你能帮卡卡吗?
输入
第一行包含一个整数N(N≤100000),这是树中的分支的数量。
下面的N - 1行每一行都包含两个整数u和v，这意味着fork u和fork v由一个分支连接。
下一行包含一个整数M(≤100000)。
以下M行每一行都包含一个消息
C x表示叉x上的苹果已经改变了。比如，如果叉子上有一个苹果，卡卡就会把它摘下来;否则，空叉子上就会长出一个新的苹果。
或
“Q x”表示在叉x上方的子树中查询苹果的数量，包括叉x上的苹果(如果存在的话)
注意一开始树里满是苹果
输出
对于每个查询，输出每行对应的答案。
解题思路：
先利用链式向前星将树结构存储起来，再用dfs获取dfs序转成线性结构，之后便可用树状数组通过dfs序对树进行维护，更改状态位置为beg【i】，求子树苹果数的线性范围为beg【i】–last【i】。

#include <iostream>
#include <cstring>
#include <stdio.h>

#define MAXSIZE 100005

using namespace std;   //使用名称空间std

int Head[MAXSIZE], Next[MAXSIZE], To[MAXSIZE];          //链式向前星（线性结构存边)
//Head[i]代表起点为i的边在向前星中最后存储的位置(头指针），To[i]表示向前星第i个位置的边的终点，Next[i]表示与第i个边相同起点的下一个边在向前星中的位置
int beg[MAXSIZE], last[MAXSIZE];                      //begin[i]和last[i]分别存储以i为起点的连通子图（树结构为子树)的dfs序的起始位置和终止位置
int cnt_edg, cnt_dfs;                                                //向前星下标(0--MAXSIZE),dfs序存储下标(1--MAXSIZE)(对应树状数组）

bool tree[MAXSIZE];           //苹果树苹果的状态(0/1)
int val[MAXSIZE];             //树状数组

int main()
{
	void add(int from, int to);
	void dfs(int fa, int son);
	int lowbit(int x);
	void update_1(int val[], int i, int cal, int arry_num);
	int sum_pre_1(int val[], int i);
	int sum_between_1(int val[], int pre, int last);
	int N, M;          //N个分叉，M个指令
	int x1, x2;
	int x;
	char order;        //指令
	while (~scanf("%d", &N)) 
	{
		memset(Head, -1, sizeof(Head));
		memset(val, 0, sizeof(val));
		cnt_edg = 0;
		cnt_dfs = 0;
		for(int i=1;i<N;i++)
		{//加边
			scanf("%d%d", &x1, &x2);
			add(x1, x2);   
		}
		dfs(-1, 1);    //获取dfs序
		for (int i = 1;i <= N;i++)
		{
			tree[i] = true;
			update_1(val, i, 1, N);
		}
		//苹果树初始化完毕
		scanf("%d", &M);
		while (M--)
		{
			getchar();     //取出缓冲区的换行符
			scanf("%c%d", &order, &x);
			if (order == 'C')
			{//改变x枝的苹果状态
				if (tree[x])
				{
					tree[x] = 0;
					update_1(val, beg[x], -1, cnt_dfs);
				}
				else
				{
					tree[x] = 1;
					update_1(val, beg[x], 1, cnt_dfs);
				}
			}
			else
				printf("%d\n", sum_between_1(val, beg[x], last[x]));
		}
	}
	return 0;
}

//链式向前星
void add(int from, int to)
{//加边(from:起点/to:终点)
	To[cnt_edg] = to;               //cnt为向前星下标计数器(第cnt条边,cnt从0开始）
	Next[cnt_edg] = Head[from];
	Head[from] = cnt_edg++;
}

void dfs(int fa, int son)
{//图遍历：fa:父结点，son:子结点   (以son为起点遍历)  (树结构是遍历son对应的子树）
 //获得一个以son为起点的dfs序()
	beg[son] = ++cnt_dfs;
	for (int i = Head[son];~i;i = Next[i])             //for循环是son的广度     ~i：i!=-1  (~非运算符，-1的补码全为1）
		if (fa != To[i])    dfs(son, To[i]);           //广度的每个子树对应一个深度
	last[son] = cnt_dfs;       //(单个深度遍历完毕)
}

//一维树状数组
int lowbit(int x)
{//返回二进制数最低位的1对应的数值
	return x & (-x);      //与运算
}

//一维树状数组
void update_1(int val[], int i, int cal, int arry_num)
{//原数组第i个元素加上cal，更新树状数组相关元素,arry_num为原数组的长度
 //可直接用于树状数组的建立
	for (;i <= arry_num;i += lowbit(i))
		val[i] += cal;
}

int sum_pre_1(int val[], int i)
{//求arry数组的前i项和
 //val为树状数组地址
	int sum = 0;
	for (;i > 0;i -= lowbit(i))       //从后向前每次跳一个lowbit
		sum += val[i];
	return sum;
}

int sum_between_1(int val[], int pre, int last)
{//求原数组arry在区间[pre-last]的和
	return sum_pre_1(val, last) - sum_pre_1(val, pre - 1);
}