CODEVS.1228 苹果树(DFS序)

To CODEVS.1228 苹果树  To poj 3321

Description

在卡卡的房子外面，有一棵苹果树。每年的春天，树上总会结出很多的苹果。卡卡非常喜欢吃苹果，所以他一直都精心的呵护这棵苹果树。我们知道树是有很多分叉点的，苹果会长在枝条的分叉点上面，且不会有两个苹果结在一起。卡卡很想知道一个分叉点所代表的子树上所结的苹果的数目，以便研究苹果树哪些枝条的结果能力比较强。

卡卡所知道的是，每隔一些时间，某些分叉点上会结出一些苹果，但是卡卡所不知道的是，总会有一些调皮的小孩来树上摘走一些苹果。

于是我们定义两种操作：

C x	表示编号为x的分叉点的状态被改变(原来有苹果的话，就被摘掉，原来没有的话，就结出一个苹果)
G x	查询编号为x的分叉点所代表的子树中有多少个苹果

我们假定一开始的时候，树上全都是苹果，也包括作为根结点的分叉1。

Input Description

第一行一个数N (n<=100000)

接下来n-1行,每行2个数u,v,表示分叉点u和分叉点v是直接相连的。

再接下来一行一个数M,(M<=100000)表示询问数

接下来M行,表示询问,询问的格式如题目所述Q x或者C x

Output Description

对于每个Q x的询问,请输出相应的结果,每行输出一个

Sample Input

3

1 2

1 3

3

Q 1

C 2

Q 1

Sample Output

3

2

思路：

　　DFS序+线段树。

代码：

 #include<cstdio>
 using namespace std;
 const int N=;
 
 int n,Enum,cnt,H[N<<],Dfn[N],Size[N],Sum[N<<];
 struct Edge
 {
     int to,nxt;
 }e[N<<];
 
 int read()
 {
     int now=;char c=getchar();
     while(c>''||c<'')c=getchar();
     while(c>=''&&c<='')now=(now<<)+(now<<)+c-'',c=getchar();
     return now;
 }
 
 void AddEdge(int u,int v)
 {
     e[++Enum].to = v;
     e[Enum].nxt = H[u];
     H[u] = Enum;
 }
 
 void dfs(int x,int fa)
 {
     Dfn[x]=++cnt;
 //    Size[x]=1;
     for(int i=H[x];i;i=e[i].nxt)
       if(e[i].to!=fa)
       {
           dfs(e[i].to,x);
 //            Size[x]+=Size[e[i].to];
             Size[x]+=Size[e[i].to]+;
       }
 }
 
 void PushUp(int rt)
 {
     Sum[rt]= Sum[rt<<]+Sum[rt<<|];
 }
 
 void Build(int l,int r,int rt)
 {
     if(l==r)
     {
         Sum[rt]=;
         return;
     }
     int m=(l+r)>>;
     Build(l,m,rt<<);
     Build(m+,r,rt<<|);
     PushUp(rt);
 }
 
 void XorPoint(int l,int r,int rt,int pos)
 {
     if(l==r)
     {
         Sum[rt]^=;
         return;
     }
     int m=(l+r)>>;
     if(pos<=m) XorPoint(l,m,rt<<,pos);
     else XorPoint(m+,r,rt<<|,pos);
     PushUp(rt);
 }
 
 int QuerySum(int l,int r,int rt,int L,int R)
 {
     if(L<=l && r<=R) return Sum[rt];
     int m=(l+r)>>,ans=;
     if(L<=m) ans+=QuerySum(l,m,rt<<,L,R);
     if(m<R) ans+=QuerySum(m+,r,rt<<|,L,R);
     return ans;
 }
 
 int main()
 {
     n=read();
     Build(,n,);
     for(int u,v,i=;i<n;i++)
       u=read(),v=read(),AddEdge(u,v),AddEdge(v,u);
     dfs(,-);
 //    for(int i=1;i<=n;i++)
 //      printf("%d:dfn:%d  size:%d\n",i,Dfn[i],Size[i]);
     int m=read();char opt[];int x;
     while(m--)
     {
         scanf("%s",opt);x=read();
         if(opt[]=='Q')
 //          printf("%d\n",QuerySum(1,n,1,Dfn[x],Dfn[x]+Size[x]-1));
           printf("%d\n",QuerySum(,n,,Dfn[x],Dfn[x]+Size[x]));
         else
           XorPoint(,n,,Dfn[x]);
     }
     return ;
 }