POJ 3321 Apple Tree

树状数组。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;
#define N 100010

int vis[N],have[N];
int low[N],high[N];
int c[N];
vector< vector<int> > edge(N);
int n,now;

int lowbit(int x)
{
    return x&(-x);
}

void init()
{
    int i;
    for(i=;i<N;i++)
    {
        have[i] = ;
        c[i] = lowbit(i);
    }
    memset(vis,,sizeof(vis));
    now = ;
    return;
}

void modify(int x)
{
    int val;
    if(have[x])
    {
        have[x] = ;
        val = -;
    }
    else
    {
        have[x] = ;
        val = ;
    }
    while(x<=n)
    {
        c[x] += val;
        x += lowbit(x);
    }
}

void dfs(int v)
{
    vis[v] = ;
    low[v] = now;
    for(int i=;i<edge[v].size();i++)
    {
        if(!vis[edge[v][i]])
        {
            dfs(edge[v][i]);
        }
    }
    high[v] = now;
    now++;
}

int sum(int x)
{
    int res = ;
    while(x>)
    {
        res += c[x];
        x -= lowbit(x);
    }
    return res;
}

int main()
{
    int i,a,b,q,v;
    char ss[];
    init();
    scanf("%d",&n);
    for(i=;i<=n-;i++)
    {
        scanf("%d%d",&a,&b);
        edge[a].push_back(b);
        edge[b].push_back(a);
    }
    dfs();
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s",ss);
        if(ss[] == 'Q')
        {
            scanf("%d",&v);
            printf("%d\n",sum(high[v])-sum(low[v]-));
        }
        else
        {
            scanf("%d",&v);
            modify(high[v]);
        }
    }
    return ;
}