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=100005;

int n,Enum,cnt,H[N<<1],Dfn[N],Size[N],Sum[N<<2];
struct Edge
{
    int to,nxt;
}e[N<<1];

int read()
{
    int now=0;char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')now=(now<<3)+(now<<1)+c-'0',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]+1;
      }
}

void PushUp(int rt)
{
    Sum[rt]= Sum[rt<<1]+Sum[rt<<1|1];
}

void Build(int l,int r,int rt)
{
    if(l==r)
    {
        Sum[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    PushUp(rt);
}

void XorPoint(int l,int r,int rt,int pos)
{
    if(l==r)
    {
        Sum[rt]^=1;
        return;
    }
    int m=(l+r)>>1;
    if(pos<=m) XorPoint(l,m,rt<<1,pos);
    else XorPoint(m+1,r,rt<<1|1,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)>>1,ans=0;
    if(L<=m) ans+=QuerySum(l,m,rt<<1,L,R);
    if(m<R) ans+=QuerySum(m+1,r,rt<<1|1,L,R);
    return ans;
}

int main()
{
    n=read();
    Build(1,n,1);
    for(int u,v,i=1;i<n;i++)
      u=read(),v=read(),AddEdge(u,v),AddEdge(v,u);
    dfs(1,-1);
//    for(int i=1;i<=n;i++)
//      printf("%d:dfn:%d  size:%d\n",i,Dfn[i],Size[i]);
    int m=read();char opt[5];int x;
    while(m--)
    {
        scanf("%s",opt);x=read();
        if(opt[0]=='Q')
//          printf("%d\n",QuerySum(1,n,1,Dfn[x],Dfn[x]+Size[x]-1));
          printf("%d\n",QuerySum(1,n,1,Dfn[x],Dfn[x]+Size[x]));
        else
          XorPoint(1,n,1,Dfn[x]);
    }
    return 0;
}