POJ 3321 Apple Tree(DFS+树状数组)

转载请注明出处，谢谢http://blog.youkuaiyun.com/ACM_cxlove?viewmode=contents    by---cxlove

 题目：给出一棵树，初始每个结点一个苹果，有修改操作，改变某个结点的状态，有查询操作，某棵子树的苹果数量

http://poj.org/problem?id=3321

首先DFS一次，给每个节点重新标号，并且给每个结点一个区间，表示他的子结点的标号区间。这样查询的话，就成了区间和了。

初始每个结点有苹果，所以树状数组的值便是能表示长度，即lowbit(i)

修改的时候，如果初始有苹果，则点更新，减1，否则点更新，加1

查询的话就是一个区间和。

一直以为这题只能树DP

#include<iostream>  
#include<cstdio>  
#include<map>  
#include<cstring>  
#include<cmath>  
#include<vector>  
#include<algorithm>  
#include<set>  
#include<string>  
#include<queue>  
#define inf 1600005  
#define M 40  
#define N 100005
#define maxn 300005  
#define eps 1e-7
#define zero(a) fabs(a)<eps  
#define Min(a,b) ((a)<(b)?(a):(b))  
#define Max(a,b) ((a)>(b)?(a):(b))  
#define pb(a) push_back(a)  
#define mp(a,b) make_pair(a,b)  
#define mem(a,b) memset(a,b,sizeof(a))  
#define LL long long  
#define MOD 2012  
#define lson step<<1
#define rson step<<1|1
#define sqr(a) ((a)*(a))  
#define Key_value ch[ch[root][1]][0]  
#define test puts("OK");  
#define pi acos(-1.0)
#define lowbit(x) ((x)&(-(x)))
#pragma comment(linker, "/STACK:1024000000,1024000000")  
using namespace std;  
struct Edge{
    int v,next;
}e[N*2];
int tot,start[N];
void add(int u,int v){
    e[tot].v=v;e[tot].next=start[u];
    start[u]=tot++;
    e[tot].v=u;e[tot].next=start[v];
    start[v]=tot++;
}
int l[N],r[N],id[N];
int n;
int idx;
int s[N];
int app[N];
void dfs(int u,int pre){
    id[u]=++idx;
    l[u]=idx;
    for(int i=start[u];i!=-1;i=e[i].next){
        int v=e[i].v;
        if(v==pre) continue;
        dfs(v,u);
    }
    r[u]=idx;
}
void Update(int x,int val){
    for(int i=x;i<=n;i+=lowbit(i))
        s[i]+=val;
}
int sum(int x){
    int ret=0;
    for(int i=x;i>0;i-=lowbit(i))
        ret+=s[i];
    return ret;
}
int main(){
    freopen("in.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        tot=0;mem(start,-1);
        for(int i=1;i<n;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);
        }
        idx=0;
        dfs(1,0);
        mem(s,0);
        for(int i=1;i<=n;i++) s[i]=lowbit(i),app[i]=1;
        int q;
        scanf("%d",&q);
        while(q--){
            char str[5];
            int k;
            scanf("%s%d",str,&k);
            if(str[0]=='C') app[k]^=1,Update(id[k],app[k]?1:-1);
            else printf("%d\n",sum(r[k])-sum(l[k]-1));
        }
    }
    return 0;
}