poj 3321 Apple Tree

方法：

树上不好直接统计和更新。

树转序列，再用数状数组。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int tree[500000],b[500000],e[500000];
bool v[500000],h[500000];
int limit;
struct node
{
    int x,next;
}a[500000];
int head[500000];
int m;
inline int lowbit(int x)
{
    return x&(-x);
}
void add(int pos,int x)
{
    while(pos<=limit)
    {
        tree[pos]+=x;
        pos+=lowbit(pos);
    }
}
int query(int pos)
{
    int sum=0;
    while(pos>0)
    {
        sum+=tree[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
void dfs(int x)//to xulie
{
    v[x]=true;
    limit++;
    b[x]=limit;
    for(int i=head[x];i!=-1;i=a[i].next)
    if(!v[a[i].x])
    {
        dfs(a[i].x);
    }
    limit++;
    e[x]=limit;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(b,0,sizeof(b));
        memset(e,0,sizeof(e));
        memset(head,-1,sizeof(head));
        m=0;
        for(int i=1;i<=n-1;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            m++;
            a[m].x=v;a[m].next=head[u];head[u]=m;
            m++;
            a[m].x=u;a[m].next=head[v];head[v]=m;
        }
        memset(v,false,sizeof(v));
        limit=0;
        dfs(1);
        memset(tree,0,sizeof(tree));
        for(int i=1;i<=n;i++)
        {
            add(b[i],1);
            h[i]=true;
        }

        int m,x;
        scanf("%d",&m);
        char c[10];
        for(int i=1;i<=m;i++)
        {
            scanf("%s%d",c,&x);
            if(c[0]=='C')
            {
                if(h[x])
                    {h[x]=false;add(b[x],-1);}
                else
                    {h[x]=true;add(b[x],1);}
            }
            else
                printf("%d\n",query(e[x])-query(b[x]-1));
        }
    }
    return 0;
}