POJ3321[苹果树]--DFS序+树状数组

【链接】
poj3321

【题目大意】
给你一颗n个节点的树，树的根节点是1，有两个操作，分别是C x表示如果x节点的权值为1则修正为0，否则修正为1。Q x表示询问以x为根节点的子树的权值。

【解题报告】
直接用DFS序将树转化为区间，然后树状数组搞一下就可以了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100005,maxm=200005;
int n,m,Time,tot,lnk[maxn],son[maxm],nxt[maxm],in[maxn],out[maxn];
bool vis[maxn];
struct BLT
{
    int c[maxn];
    void Clear(){memset(c,0,sizeof(c));}
    int Lowbit(int x){return x&(-x);}
    void Add(int x,int da){while (x<=n) c[x]+=da,x+=Lowbit(x);}
    int Get(int x){int sum=0; while (x) sum+=c[x],x-=Lowbit(x); return sum;}
}tr;//树状数组
inline int Read()
{
    int res=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=getchar();
    return res;
}
void Dfs(int x,int fa)
{
    in[x]=++Time;
    for (int j=lnk[x]; j; j=nxt[j])
     if (son[j]!=fa) Dfs(son[j],x);
    out[x]=Time;
}
void Add_e(int x,int y)
{
    son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
}
int main()
{
    freopen("3321.in","r",stdin);
    freopen("3321.out","w",stdout);
    n=Read(); Time=0;
    for (int i=1,x,y; i<n; i++) x=Read(),y=Read(),Add_e(x,y),Add_e(y,x);
    Dfs(1,0); m=Read(); tr.Clear();
    for (int i=1; i<=n; i++) tr.Add(i,1);
    memset(vis,0,sizeof(vis));
    for (int i=1; i<=m; i++)
    {
        char ch=getchar();
        while (ch!='Q'&&ch!='C') ch=getchar();
        int x=Read();
        if (ch=='Q') printf("%d\n",tr.Get(out[x])-tr.Get(in[x]-1));
         else {if (vis[x]) tr.Add(in[x],1); else tr.Add(in[x],-1); vis[x]=1-vis[x];}
    }
    return 0;
}