POJ 3321 Apple Tree（dfs序+线段树）

POJ 3321 Apple Tree （题目链接）

题意：给出一个苹果树，每个节点一开始都有苹果
C X，如果X点有苹果，则拿掉，如果没有，则新长出一个
Q X，查询以X点为根的子树一共有几个苹果

思路：利用dfs将树上的节点hash成线性，并将子树hash成区间。然后就转换成裸线段树了。

/*************************************************************************
     File Name: A.cpp
     ID: obsolescence
     BLOG: http://blog.csdn.net/obsolescence
     LANG: C++ 
     Mail: 384099319@qq.com 
     Created Time: 2016年07月28日 星期四 10时58分42秒
 ************************************************************************/
//#include<bits/stdc++.h>
#include<cstdio>
#include<cstring>
#define Max(x,y) ((x)>(y)?(x):(y))
#define Min(x,y) ((x)<(y)?(x):(y))
#define each(it,v) for(__typeof((v).begin()) it=(v).begin();it!=(v).end();++it)
#define Abs(x,y) ((x)>(y)?((x)-(y)):((y)-(x)))
#define ll long long
#define Mem0(x) memset(x,0,sizeof(x))
#define Mem1(x) memset(x,-1,sizeof(x))
#define MemX(x) memset(x,0x3f,sizeof(x))
#define pb push_back
#define left t<<1
#define right t<<1|1
using namespace std;
const int N=100010;
struct Edge{
  int to;
  Edge *next;
}*head[N],e[N<<1];
int hash1[N],hash2[N],top,tot,ans,tree[N<<2];

void AddEdge(int from,int to) {
  Edge *p=&e[++top];
  p->to=to,p->next=head[from],head[from]=p;
}

void dfs(int ch,int pre) {
  ++tot;
  hash1[ch]=tot;//x的序号
  for (Edge *p=head[ch]; p; p=p->next) {
    if (p->to!=pre) dfs(p->to,ch);
  }
  hash2[ch]=tot;//以x为根的子树的最后一个节点的序号，查询以x为根的子树即为查询区间[hash1,hash2]
}

void pushUp(int t) {
  tree[t]=tree[left]+tree[right];
}

void build(int t,int L,int R) {
  if (L==R) {
    tree[t]=1;
    return;
  }
  int mid=(L+R)>>1;
  build(left,L,mid);
  build(right,mid+1,R);
  pushUp(t);
}

void update(int t,int L,int R,int x) {
  if (L==R) {
    tree[t]^=1;
    return;
  }
  int mid=(L+R)>>1;
  if (x<=mid) update(left,L,mid,x);
  if (x>mid) update(right,mid+1,R,x);
  pushUp(t);
}

void query(int t,int L,int R,int l,int r) {
  if (l<=L && R<=r) {
    ans+=tree[t];
    return;
  }
  int mid=(L+R)>>1;
  if (l<=mid) query(left,L,mid,l,r);
  if (r>mid) query(right,mid+1,R,l,r);
}

int main() {
  int n,m,i,p1,p2,x;
  char op[2];
  while (~scanf("%d",&n)) {
    Mem0(head),top=0,tot=0;
    for (i=1; i<n; ++i) {
      scanf("%d%d",&p1,&p2);
      AddEdge(p1,p2);
      AddEdge(p2,p1);
    }
    dfs(1,-1);
    build(1,1,n);
    scanf("%d",&m);
    while (m--) {
      scanf("%s%d",op,&x);
      if (op[0]=='Q') {
        ans=0;
        query(1,1,n,hash1[x],hash2[x]);
        printf("%d\n",ans);
      } else {
        update(1,1,n,hash1[x]);
      }
    }
  }
}