POJ 3321

【题目分析】
dfs序+树状数组

---

【代码】

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int in[100001];
int out[100001];
int cnt=1;
int h[200001],ne[200001],to[200001],en=0;
int t[200001];
int n,q;
char s[11];
int a[200001];
inline void add(int a,int b)
{ne[en]=h[a];to[en]=b;h[a]=en++;}
inline void dfs(int k,int & cnt)
{
    in[k]=cnt;
    for (int i=h[k];i>=0;i=ne[i])
        if (!in[to[i]]) dfs(to[i],++cnt);
    out[k]=cnt;
}
inline void addup(int k)
{for (;k<=n;k+=k&(-k)) t[k]++;}
inline void dec(int k)
{for (;k<=n;k+=k&(-k)) t[k]--;}
inline int getsum(int k)
{
    int ret=0;
    for (;k>=1;k-=k&(-k)) ret+=t[k];
    return ret;
}
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    for (int i=1;i<n;++i)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    dfs(1,cnt);
    for (int i=1;i<=n;++i) addup(i),a[i]=1;
    scanf("%d",&q);
    for (int i=1;i<=q;++i)
    {
        int x;
        scanf("%s",s+1);
        scanf("%d",&x);
        if (s[1]=='Q') printf("%d\n",getsum(out[x])-getsum(in[x]-1));
        if (s[1]=='C') {if (!a[in[x]]) addup(in[x]),a[in[x]]=1; else dec(in[x]),a[in[x]]=0;}
    }
}