题目,是对一颗树,单点修改、子树查询。典型的dfs序入门题。
DFS序可以将一颗树与子树们表示为一个连续的区间,然后用线段树来维护;感觉算是树链剖分的一种吧,和轻重链剖分不同的是这是对子树进行剖分的。
我用非递归方式求DFS序。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 111111
struct Edge{
int v,nxt;
}edge[MAXN<<];
int n,NE,head[MAXN];
void addEdge(int u,int v){
edge[NE].v=v; edge[NE].nxt=head[u]; head[u]=NE++;
} int f[MAXN],l[MAXN],r[MAXN],stack[MAXN],odr;
void dfs(){
int top=;
stack[++top]=;
while(top){
int u=stack[top];
if(l[u]){
r[u]=odr; --top;
continue;
}
l[u]=++odr;
for(int i=head[u]; i!=-; i=edge[i].nxt){
int v=edge[i].v;
if(f[u]==v) continue;
f[v]=u; stack[++top]=v;
}
}
} int N,tree[MAXN<<],x,y;
void update(int i,int j,int k){
if(i==j){
tree[k]^=;
return;
}
int mid=i+j>>;
if(x<=mid) update(i,mid,k<<);
else update(mid+,j,k<<|);
tree[k]=tree[k<<]+tree[k<<|];
}
int query(int i,int j,int k){
if(x<=i&&j<=y) return tree[k];
int mid=i+j>>,res=;
if(x<=mid) res=query(i,mid,k<<);
if(y>mid) res+=query(mid+,j,k<<|);
return res;
} int main(){
int m,a,b;
char op[];
memset(head,-,sizeof(head));
scanf("%d",&n);
for(int i=;i<n;++i){
scanf("%d%d",&a,&b);
addEdge(a,b);
addEdge(b,a);
}
dfs();
for(N=; N<odr; N<<=);
for(int i=; i<=n; ++i){
x=l[i];
update(,N,);
}
scanf("%d",&m);
while(m--){
scanf("%s%d",op,&a);
if(op[]=='Q'){
x=l[a]; y=r[a];
printf("%d\n",query(,N,));
}else{
x=l[a];
update(,N,);
}
}
return ;
}