线段树 POJ 3321 Apple Tree

题目链接：http://poj.org/problem?id=3321

代码风格：http://www.notonlysuccess.com/index.php/segment-tree-complete/

题目大意：Q——问在A苹果的支上有多少个苹果，C——把A苹果吃掉

算法：dfs 线段树 邻接表

思路：先建好邻接关系， 如图建立由这些节点的位置组成的线段树，然后求两次经过该节点的两个数之间有有多少个苹果（红色表示x号苹果，蓝色左边的表示第一次经过该节点记录下的位置，右边表示第二次经过节点记录下的位置）,如图，build(1, 22,1);把每个节点左边位置加入，也就是一共加入11次数字了，然后再进行删数求和操作

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1
#define mid int m = (l+r) >> 1

int tot = 0;
int sum[565656], head[565656], vis[565656], li[565656], ri[565656];
struct nod
{
	int v, next ;
}edge[565656];
stack<int> s;

void add_edge(int s, int v)
{
	edge[tot].v = v;
	edge[tot].next = head[s];
	head[s] = tot ++;
}

void dfs(int x)
{
	while(!s.empty())
		s.pop();
	s.push(x);
	int cnt = 1;
	while(!s.empty())
	{
		int now = s.top();
//		cout << "now = " << now << endl;
		if(!vis[now])
		{
			li[now] = cnt ++;
			vis[now] = 1;
		}
		int f = 0;
		for(int i = head[now]; i != -1; i = edge[i].next)
		{
		//	cout << "i = " << i << endl;
			if(!vis[edge[i].v])
			{
				s.push(edge[i].v);
				f = 1;
				head[now] = edge[i].next;
				break;
			}
		}
		if(f)
			continue;
		if(vis[now])
		{
			ri[now] = cnt ++;
			s.pop();
		}
	}
}

void PushUp(int rt)
{
	sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}

void update(int k, int l, int r, int rt)
{
	if(l == r)
	{
		sum[rt] ^= 1;
		return ;
	}
	mid ;
	if(k <= m)
		update(k, lson);
	else update(k, rson);
	PushUp(rt);
}

int query(int L, int R, int l, int r, int rt)
{
	if(L <= l && r <= R)
	{
		return sum[rt];
	}
	mid ;
	int ret = 0;
	if(L <= m)
		ret += query(L, R, lson);
	if(m < R)
		ret += query(L, R, rson);
	return ret ;
}



int main(){
	int n;
	int a, b, m;
	char op[50];
	scanf("%d", &n) ;
		int i;
		memset(vis, 0, sizeof(vis));
		memset(head, -1, sizeof(head));
		memset(sum, 0, sizeof(sum));
		for(i = 1; i < n; i ++)
		{
			scanf("%d%d", &a, &b);
			add_edge(a, b);
			add_edge(b, a);
		}
		dfs(1);
		for(i = 1; i <= n; i ++)
		{
			update(li[i], 1, n * 2, 1);
		}
		scanf("%d", &m);
		while(m --)
		{
			scanf("%s%d", op, &a);
			if(op[0] == 'Q')
				printf("%d\n" , query(li[a], ri[a], 1, n << 1, 1));
			else update(li[a], 1, n << 1, 1);
		}
	return 0;
}