POJ

原题链接

题意：

给你一个$n$个节点的树，每个节点上一开始都有一个苹果，现在有两种操作：

$Q_{x}$ ：查询 $x$ 节点以及它的子树上一共有多少苹果；

$C_{x}$ ：若 $x$ 节点上没有苹果，则增加一个，反之，则减少一个；

思路：

这是一个区间查询，单点修改的问题，我们就会想到用树状数组来做，但是怎么去维护树上的区间又是一个问题；

于是想到了用 dfs序 ，记录每一个节点的 left 和 right 也就转化成了区间问题；

/*
* @Author: windystreet
* @Date:   2018-08-02 22:10:44
* @Last Modified by:   windystreet
* @Last Modified time: 2018-08-02 22:56:03
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>

using namespace std;

#define X first
#define Y second
#define eps  1e-2
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!
");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 1e5+7;
const LL  INF  = 1<<30;
const int mod  = 1e9+7;
 typedef vector<int> Ve;
vector<Ve>G(maxn);
int left[maxn],right[maxn],cnt=1,n;
int tree[maxn],vis[maxn];

int query(int x){
    int res = 0;
    while(x){
        res+=tree[x];
        x-=lowbit(x);
    }
    return res;
}
void add(int x,int v){
    for(int i=x;i<=n;i+=lowbit(i)){
        tree[i]+=v;
    }
    return ;
}
void dfs(int x){
    left[x] = cnt;                     // 记录左端点的id
    int sz = G[x].size();
    for(int i=0;i<sz;i++){
        cnt++;
        dfs(G[x][i]);
    }
    right[x]=cnt;                      // 记录右端点的id
}

void solve()
{
    cnt = 1;
    int x,y,m;
    char op[3];
    for(int i=0;i<=n;i++){tree[i]=0;vis[i]=1;G[i].clear();}
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        G[x].pb(y);                     // 存图
    }
    dfs(1);                             // dfs序
    for(int i=1;i<=n;i++){
        add(i,1);                       // 每个点一开始都有一个苹果
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%s%d",op,&x);
        if(op[0]=='Q'){                 // 区间查询
            printf("%d
",query(right[x])-query(left[x]-1));
        }else{
            if(vis[x])add(left[x],-1);  // 单点修改
            else add(left[x],1);
            vis[x]=!vis[x];             // 取反
        }
    }
    return ;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    while(scanf("%d",&n)!=EOF){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}