poj3321

求树上以i节点为跟的子树的权值和，支持点更新操作

关键词：tree_to_line、树状数组求区间和

做法：dfs将树上的顶点按照第一次的遍历时间顺序放入到数组中，则由dfs性质可得：以i为跟的子树所有定点在数组中是一个连续序列。记录i子树在数组中的起点st[i]和终点en[i]，对于询问子树权值的问题转化为求连续区间和的问题---树状数组！

树状数组的应用之一：O(logn)时间内求连续区间的和(len=n)

#include<stdio.h>
#include<iostream>
#include<algorithm>
#define maxn 100100
#define mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
using namespace std;

int n,m,cnt;
int c[maxn],st[maxn],en[maxn],flag[maxn];
struct node{
    int to,next;
}edge[maxn*2];
int head[maxn],tot;

void add(int u,int v){
    edge[tot].to=v,edge[tot].next=head[u];
    head[u]=tot++;
}

void update(int i,int j){
    for(;i<=n;i+=lowbit(i)) c[i]+=j;
}

int sum(int i){
    int s=0;
    for(;i>0;i-=lowbit(i)) s+=c[i];
    return s;
}

void dfs(int u,int f){
   st[u]=++cnt;//记录深搜访问的顺序cnt
   update(cnt,1);
   for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v!=f) dfs(v,u);
   }
   en[u]=cnt;
}

int main(){
    //freopen("a.txt","r",stdin);
    scanf("%d",&n);
    mem(c,0),mem(head,-1),mem(flag,0),tot=0;
    for(int i=0;i<n-1;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    cnt=0;
    dfs(1,0);
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        char a[5];
        int b;
        scanf("%s%d",a,&b);
        if(a[0]=='Q') printf("%d\n",sum(en[b])-sum(st[b]-1));
        else{
            if(!flag[st[b]]) { update(st[b],-1); flag[st[b]]=1; }
            else { update(st[b],1); flag[st[b]]=0; }
        }
    }
    return 0;
}