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