poj3321 树状数组

每个节点有权值为0/1，初始化为1，每次操作都使得某点权值变化，每次询问某点及其子树权值和。

树的子树权值和问题可转化为线性数组问题：记st[u]为dfs第一次访问u的时间戳，en[u]为dfs回溯到u的时间戳。

从而子树权值和问题转化为连续区间求和和点更新问题，可以用树状数组进行解决。

变题：每个点权值初始化为任意值；点更新时可以更新为任意值，求子树权值和。

思路类似，update的数值修改即可。

#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;
}