POJ3321 Apple Tree（DFS序）

题目，是对一颗树，单点修改、子树查询。典型的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 ;
 }