POJ 3321 Apple Tree (树状数组)

思路：用了时间戳，什么是时间戳？就是用begin[i]记录以i为根的，开始遍历的序号，end[i]记录以i为根，最后遍历的序号。

如下挫图：

从1开始遍历是，他是第一个遍历的，所以标为1，之后遍历所有子树后，到5的时候，5是第4个遍历的，所以5的begin值为4，而返回后，1的所有子树最大遍历标号就是5的begin值，所以2的end值为4，同理，1的end值为3的begin值为5

那么，可以发现，对于某个节点，我们查询他苹果数的时候，他所占的区域其实就是begin[i]~end[i]，这样我们用树状数组求出sum(end[i])-sum(begin[i]-1)即可(当然包括begin[i])，对于更新，只要更新add(begin[i])就行了。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=100005;
int c[maxn],index,n;
int head[maxn],edge,begin[maxn],end[maxn];
int to[maxn<<1],next[maxn<<1];
bool flag[maxn],vis[maxn];
inline void addedge(int u,int v)
{
    to[edge]=v,next[edge]=head[u],head[u]=edge++;
}
void init()
{
    for(int i=1; i<=n; i++)
        head[i]=-1,vis[i]=0;
    edge=index=0;
}
void add(int x)
{
    int val;
    if(flag[x])
    {
        val=-1;
        flag[x]=0;
    }
    else val=1,flag[x]=1;
    while(x<=n)
        c[x]+=val,x+=(x&-x);
}
int sum(int x)
{
    int s=0;
    while(x>0)
        s+=c[x],x-=(x&-x);
    return s;
}
void dfs(int now)
{
    begin[now]=++index;
    vis[now]=1;
    for(int i=head[now]; ~i; i=next[i])
        if(!vis[to[i]])
            dfs(to[i]);
    end[now]=index;
}
int main()
{
    int i,u,v,m;
    char ch[2];
    while(~scanf("%d",&n))
    {
        init();
        memset(c,0,sizeof(c));
        for(i=0; i<n-1; ++i)
        {
            scanf("%d%d",&u,&v);
            addedge(u,v);
            addedge(v,u);
        }
        for(i=1; i<=n; i++)
        {
            add(i);
            flag[i]=1;
        }
        dfs(1);
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",ch);
            if(ch[0]=='Q')
            {
                scanf("%d",&u);
                printf("%d\n",sum(end[u])-sum(begin[u]-1));
            }
            else
            {
                scanf("%d",&u);
                add(begin[u]);
            }
        }
    }
    return 0;
}