POJ 3321 Apple Tree （树状数组）

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

题意：给出一颗树，一开始树是满的（每个结点都有苹果）两种操作；1）C x ，x结点有苹果则摘掉这个苹果，否则加上一个苹果；2）Q x ，询问结点x为根的子树上共有多少苹果（包括x上的苹果）

思路：

关键是把这颗树转化为树状数组，一颗子树中的结点要占连续的空间，访问时才方便方法是dfs给各个结点编号。

每个结点对应两个编号：以这个结点为跟的子树所占的连续空间的开始位置s和结束位置e。

树状数组bit[ i ]记录从开头到 i 这一段的和，那么所求就是bit[ e ] - bit[ s ]。

才知道POJ用stl会很慢啊（大概是指针操作的问题吧），要用数组保存边信息建树，用vector会T。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <stdlib.h>
#define INF 0x7fffffff
#define MOD 1000000007
#include <vector>
using namespace std;
typedef long long ll;
const int MAXN = 100005;
vector<int> G[MAXN];
int n, m;
int s[MAXN], e[MAXN], cnt, apple[MAXN], bit[2 * MAXN];

struct Edge
{
int to, next;
}edge[2 * MAXN];
int f[MAXN];
void addedge(int u, int v)
{
edge[cnt].to = v;
edge[cnt].next = f[u];
f[u] = cnt++;
}
void dfs(int u)
{
s[u] = cnt++;
for(int i = f[u]; i != -1; i = edge[i].next)
{
if(!s[edge[i].to])
dfs(edge[i].to);
}
e[u] = cnt++;
}

void update(int i, int x)
{
//printf("update %d\n", i);
while(i <= 2 * n)
{
bit[i] += x;
i += i & -i;
}
}

int sum(int i)
{
int s = 0;
while(i > 0)
{
s += bit[i];
i -= i & -i;
}
return s;
}

int main()
{
#ifdef LOCAL
freopen("data.in", "r", stdin);
#endif

int u, v, x;
char c;
while(scanf("%d", &n) != EOF)
{
for(int i = 0; i <= n; i++)
G[i].clear();
memset(bit, 0, sizeof(bit));
memset(s, 0, sizeof(s));
memset(e, 0, sizeof(e));
memset(apple, 0, sizeof(apple));

memset(f, -1, sizeof(f));
cnt = 0;
for(int i = 0; i < n - 1; i++)
{
scanf("%d%d", &u, &v);
//G[u].push_back(v); G[v].push_back(u);
addedge(u, v);
addedge(v, u);
}

cnt = 1;
dfs(1);
for(int i = 1; i <= n; i++)
{
//printf("%d:%d-%d\n", i, s[i], e[i]);
update(e[i], 1);
apple[i] = 1;
}

scanf("%d", &m);
for(int i = 0; i < m; i++)
{
scanf(" %c%d", &c, &x);
//printf("%c %d\n", c, x);
if(c == 'Q')
{
printf("%d\n", sum(e[x]) - sum(s[x]));
}
else
{
if(apple[x])
{
update(e[x], -1);
apple[x] = 0;
}
else
{
update(e[x], 1);
apple[x] = 1;
}
}
}

}

return 0;
}