DFS序就是将树形结构转化为线性结构,用dfs遍历一遍这棵树,进入到x节点有一个in时间戳,递归退出时有一个out
时间戳,x节点的两个时间戳之间遍历到的点,就是根为x的子树的所有节点,他们的dfs进入时间戳是递增的。同时两个时间戳构成了一个区间,x节点在这段区间的最左端,这个区间就是一棵根节点为x的子树,对于区间的操作就是其他维护方式的应用了。
int time = 0;
inline void dfs(int x, int fa) {
in[x] = ++time; //进入的时间戳
num[time] = x; //生成新的线性结构
for(int i = 0; i < G[x].size(); i++) {
int cnt = G[x][i];
if(cnt == fa) continue;
dfs(cnt, x);
}
out[x] = time; //出去的时间戳
}
例题 poj 3321
题意:
给出一个苹果树,每个节点一开始都有苹果
C X,如果X点有苹果,则拿掉,如果没有,则新长出一个
Q X,查询X点与它的所有后代分支一共有几个苹果
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
const int N=1e5+10;
int n;
int head[N<<2];
struct node
{
int to,next;
}edge[N<<2];
int cc;
void addedge(int from,int to)
{
edge[cc].to=to;
edge[cc].next=head[from];
head[from]=cc++;
}
vector<vector<int> >vec(N);
int c[N];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int val)
{
while(x<=n){
c[x]+=val;
x+=lowbit(x);
}
}
int sum(int x)
{
int res=0;
while(x>0){
res+=c[x];
x-=lowbit(x);
}
return res;
}
int cnt;
int in[N];
int out[N];
void dfs(int u)
{
in[u]=++cnt;
for(int i=head[u];i!=-1;i=edge[i].next){
int to=edge[i].to;
dfs(to);
}
out[u]=cnt;
}
int vis[N];
void Init()
{
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
memset(in,0,sizeof(in));
memset(out,0,sizeof(out));
for(int i=0;i<=n;i++) vec[i].clear();
cc=0;
cnt=0;
}
int main()
{
while(scanf("%d",&n)==1){
Init();
for(int i=0;i<n-1;i++){
int a,b;
scanf("%d %d",&a,&b);
addedge(a,b);
}
int m;
scanf("%d",&m);
dfs(1);
for(int i=1;i<=n;i++){
add(i,1);
vis[i]=1;
}
char s[10];
while(m--){
int x;
scanf("%s %d",s,&x);
if(s[0]=='Q'){
printf("%d\n",sum(out[x])-sum(in[x]-1));
}
else{
if(vis[x]){
add(in[x],-1);
}
else{
add(in[x],1);
}
vis[x]=!vis[x];
}
}
}
return 0;
}