【Codevs1228】苹果树（dfs序+树状数组）

#include
#include
#include
#include
using namespace std;
const int maxn=100010;
int n,m,a1,a2,in[maxn],out[maxn],head[maxn],ncnt,cnt,c[maxn],vis[maxn];
struct edge{
    int to,ne;
}e[maxn];

inline int lowbit(int x){return x&(x^(x-1));}

void add(int x,int y){
    e[++ncnt].to=y;
    e[ncnt].ne=head[x];
    head[x]=ncnt;
}

void dfs(int x){//深搜生成dfs序 
    in[x]=++cnt;//进的时间戳 
    for(int i=head[x];i;i=e[i].ne){
        dfs(e[i].to);
    }
    out[x]=cnt;//出的时间戳 
}

void change(int x,int k){
    while(x<=n){
        c[x]+=k;
        x+=lowbit(x);
    }
}

int query(int x){
    int ans=0;
    while(x){
        ans+=c[x];
        x-=lowbit(x); 
    }
    return ans;
}

void init(){
    for(int i=1;i<=n;i++){
        change(in[i],1);
        vis[i]=1;//每一个节点都有苹果 
    }
}

int main(){
    cin>>n;
    for(int i=1;iscanf("%d%d",&a1,&a2);
        add(a1,a2);
    }
    vis[n]=1;
    dfs(1);
    init();
    cin>>m;
    while(m--){
        char ch;
        int x;
        cin>>ch;
        cin>>x;
        if(ch=='C'){
            if(vis[x]){
                change(in[x],-1);
            }else{
                change(in[x],1);
            }
            vis[x]=!vis[x];
        }else{
            printf("%d\n",query(out[x])-query(in[x]-1));
        }
    }
    return(0);
}