求树上以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;
}