codevs 1228 苹果树 题解报告

题目描述 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

数据范围及提示 Data Size & Hint

看数据范围可以看出是一道线段树的题目。

要求在树上单点修改 区间查询。
所以先跑 DFS
求 DFS序
在DFS序上跑线段树

记录 DFN[] DFL[]
分别表示 以该点为根的子树起点编号和中点编号
之后就是一个普通的线段树了。

代码：：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<cstdlib>
#include<string>
#include<bitset>
#include<iomanip>
#include<deque>
#define INF 1000000000
#define fi first
#define se second
#define N 100005
#define P 1000000007
#define debug(x) cerr<<#x<<"="<<x<<endl
#define MP(x,y) make_pair(x,y)
using namespace std;
int n,m,ti=0,dfn[100010],dfl[100101],sum;
int p=0,h[100011],t[500010];
struct zqm
{
    int v,next,w;
}q[200001];
void add(int a,int b)
{
    p++;
    q[p].v=b;
    q[p].next=h[a];
    h[a]=p;
}
inline int get_num()
{
int num = 0;
char c;
bool flag = false;
while ((c = getchar()) == ' ' || c == '\n' || c == '\r');
if (c == '-') flag = true;
else num = c - '0';
while (isdigit(c = getchar()))
num = num * 10 + c - '0';
return (flag ? -1 : 1) * num;
}
void dfs(int x)
{
    ti++;
    if(!dfn[x])
    {
        dfn[x]=ti;
    }
    int pp=h[x];
    while(pp)
    {
        int k=q[pp].v;
        if(!dfn[k])
        {
            dfs(k);
        }
        pp=q[pp].next;
    }
    dfl[x]=ti;
}
void build(int x,int l,int r)
{
    if(l==r)
    {
        t[x]=1;
        return;
    }
    int mid=(l+r)>>1;
    build(2*x,l,mid);
    build(2*x+1,mid+1,r);
    t[x]=t[2*x]+t[2*x+1];
}
void chenge(int x,int l,int r,int e)
{
    if(l==r)
    {
        if(l==dfn[e])
        {
            if(t[x]==1)
            {
                t[x]=0;
            }
            else
            {
                t[x]=1;
            }
        }
        return;
    }
    int mid=(l+r)>>1;
    if(mid>=dfn[e])
    {
        chenge(2*x,l,mid,e);
    }else
    {
        chenge(2*x+1,mid+1,r,e);
    }
    t[x]=t[2*x]+t[2*x+1];
}
void find(int x,int l,int r,int w,int e)
{
    if(l==w&&r==e)
    {
        sum+=t[x];
        return;
    }
    int mid=(l+r)>>1;
    if(mid>=e)
    {
        find(2*x,l,mid,w,e);
    }else
    {
        if(mid<w)
        {
            find(2*x+1,mid+1,r,w,e);
        }else
        {
            find(2*x,l,mid,w,mid);
            find(2*x+1,mid+1,r,mid+1,e);
        }
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int q,w;
        cin>>q>>w;
        add(q,w);
        add(w,q);
    }
    dfs(1);
    /*for(int i=1;i<=n;i++)
    {
        cout<<dfn[i]<<" "<<dfl[i]<<endl;
    }*/
    build(1,1,n);
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        char c;
        int e;
        cin>>c;
        if(c=='C')
        {
            cin>>e;
            chenge(1,1,n,e);
        }else
        {
            cin>>e;
            sum=0;
            find(1,1,n,dfn[e],dfl[e]);
            cout<<sum<<endl;
        }
    }
}