POJ 3321 Apple Tree（ 树状数组）

题目大意：
给你一颗苹果树，树的主干设为1，每一个分支设为一个数，一直到N，代表这颗苹果树。每个分支上面只能最多有一个苹果，也就是一个枝子上面不可能有两个苹果。
输入是叉之间的关系(数据是根结点向叶结点的方向)
1 2
1 3
就是主干上面两个叉分别是2 和3.
下面是两种操作，Q 和C
C   j  的意思是如果 j 这个枝子上面有苹果就摘下来，如果没有，那么就会长出新的一个

Q  j  就是问 j 这个叉的子树上面的苹果总数。

思路：修改节点的苹果数，然后查询某一子树（区间）苹果的苹果总数，显然可以用线段树和树状数组做

这题的难点是如何把子树映射成对应的区间，也不难想到可以用dfs遍历整个树，时间戳给这些节点确定好了关系，记录某一个节点的时间戳和回溯的时间戳，那么以它为根的子树的区间就确定下来了

//4244K	360MS
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define lowbit(x) (x&-x)
#define MAXN 100100
struct node {
    int to,next;
}es[MAXN];
int head[MAXN];
int C[MAXN];
int n,m,cnt,time;
bool vis[MAXN],has[MAXN];
int st[MAXN],ed[MAXN];
void add_edge(int u,int v){
    es[cnt].to=v;
    es[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int x){
    vis[x]=true;
    st[x]=++time;
    for(int k=head[x];k!=-1;k=es[k].next){
        if(!vis[es[k].to ]){
            dfs(es[k].to);
        }
    }
    ed[x]=time;
}
void update(int p,int val){
    for(int i=p;i>0;i-=lowbit(i)){
        C[i]+=val;
    }
}
int sum(int p){
    int s=0;
    for(int i=p;i<=n;i+=lowbit(i)){
        s+=C[i];
    }
    return s;
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add_edge(a,b);
    }
    dfs(1);
    for(int i=1;i<=n;i++){
        update(i,1);
    }
    fill(has,has+1+n,1);
    scanf("%d",&m);
    while(m--){
        char op[5];
        int x;
        scanf("%s%d",op,&x);
        if(op[0]=='C'){
            if(has[x]){
                update(st[x],-1);
                has[x]=0;
            }
            else {
                update(st[x],1);
                has[x]=1;
            }
        }
        else {
            printf("%d\n",sum(st[x])-sum(ed[x]+1));
        }
    }
    return 0;
}

以上是对x递增路求和，当然也可以写成易理解的形式：

//4244K	407MS
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define lowbit(x) (x&-x)
#define MAXN 100100
struct node {
    int to,next;
}es[MAXN];
int head[MAXN];
int C[MAXN];
int n,m,cnt,time;
bool vis[MAXN],has[MAXN];
int st[MAXN],ed[MAXN];
void add_edge(int u,int v){
    es[cnt].to=v;
    es[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int x){
    vis[x]=true;
    st[x]=++time;
    for(int k=head[x];k!=-1;k=es[k].next){
        if(!vis[es[k].to ]){
            dfs(es[k].to);
        }
    }
    ed[x]=time;
}
void update(int p,int val){
    for(int i=p;i<=n;i+=lowbit(i)){
        C[i]+=val;
    }
}
int sum(int p){
    int s=0;
    for(int i=p;i>0;i-=lowbit(i)){
        s+=C[i];
    }
    return s;
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add_edge(a,b);
    }
    dfs(1);
    for(int i=1;i<=n;i++){
        update(i,1);
    }
    fill(has,has+1+n,1);
    scanf("%d",&m);
    while(m--){
        char op[5];
        int x;
        scanf("%s%d",op,&x);
        if(op[0]=='C'){
            if(has[x]){
                update(st[x],-1);
                has[x]=0;
            }
            else {
                update(st[x],1);
                has[x]=1;
            }
        }
        else {
            printf("%d\n",sum(ed[x])-sum(st[x]-1));
        }
    }

    return 0;
}