POJ3321

题目很简单，有一颗苹果树，每个节点有一个苹果，两种操作，

1  C  i   去看第 i 个节点， 如果有苹果摘掉，没有就按上一个。

2  Q  i  查询 i 子树上总苹果数目。。。

这题感觉可以树剖，。。。 但是 dfs 序 好写。。。

ls 记录起点， rs 记录终点， 之后记录下 dfs 序。树状数组求和，，，完美。。。

 

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int maxn=100005;
const int low(int x){return x&-x;}
int num[maxn],a[maxn];
inline void add(int x,int k){for(int i=x;i<maxn;i+=low(i))num[i]+=k;}
inline int get(int x){int sum=0;for(int i=x;i;i-=low(i))sum+=num[i];return sum;}
inline int sum(int l,int r){return get(r)-get(l-1);}
int ls[maxn],rs[maxn],cnt;
struct edge{int to,nxt;}ed[maxn];
int head[maxn],tot;
inline void addedge(int a,int b){ed[++tot].to=b;ed[tot].nxt=head[a];head[a]=tot;}
void dfs(int s,int p)
{
    ls[s]= cnt++;
    for(int i=head[s];i;i=ed[i].nxt)
    {
        int t = ed[i].to;
        if(t == p)continue;
        dfs(t, s);
    }
    rs[s] = cnt-1;
}
inline void init(int n)
{
    memset(num, 0, sizeof num);
    memset(ls, 0, sizeof ls);
    memset(rs, 0, sizeof rs);
    memset(head, 0, sizeof(head));
    cnt = 1, tot = 0;
    for(int i=1; i<=n;i++)
    a[i] = 1, add(i, 1);
}
int main()
{
    int n; cin>>n;
    init(n);
    for(int i=1;i<n;i++)
    {
        int u,v;cin>>u>>v;
        addedge(u, v);
    }
    dfs(1, 1);
    int q,v; cin>>q;
    char op[10];
    while(q--)
    {
        cin>>op;cin>>v;
        if(op[0] == 'C')
        {
            if(a[ls[v]] == 1)
            {
                a[ls[v]] = 0;
                add(ls[v], -1);
            }
            else
            {
                a[ls[v]] = 1;
                add(ls[v], 1);
            }
        }
        else
        {
            cout<<sum(ls[v], rs[v])<<endl;
        }
    }
    return 0;
}