[Codevs] 1228 苹果树

1228 苹果树

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 钻石 Diamond

 

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

分析 Analysis

首先这道题理论基础：DFS序（树操）+ 线段树

好在那个长苹果不是每秒，不然我就ctrl+w了qwq

首先读进来后撸成DFS序为基础的线性结构

那么根据DFS序，这道题可以化归为：树上单点修改 + 求子树和

如果到这里还没看懂请补全DFS序的理论部分qwq（然后怎么做超显然的）

-------------------------------------------------------------------------

我第一次拿到这道题的时候，还真不知道怎么写

好在CZL夏令营的时候帮我补了树操

orzCZL

代码 Code

#include<cstdio>
#include<iostream>
#define mid (L+R)/2
#define lc (rt<<1)
#define rc (rt<<1|1)
#define maxn 1000000
using namespace std;

int dfn[maxn],TIM = 1,dfn_t[maxn],n,a,b,c,m,pos,tl[maxn];
char ctr;

struct edge{
    int from,v;
}e[maxn];

struct node{
    int sum;
}Tree[maxn];

void maintain(int rt){
    Tree[rt].sum = Tree[lc].sum+Tree[rc].sum;
}

void build(int rt,int L,int R){
    if(L == R) Tree[rt].sum = 1;
    else{
        build(lc,L,mid);
        build(rc,mid+1,R);
        
        maintain(rt);
    }
}

void add(int rt,int L,int R,int pos){
    if(L == R) Tree[rt].sum ^= 1;
    else{
        if(pos <= mid) add(lc,L,mid,pos);
        else add(rc,mid+1,R,pos);
        
        maintain(rt);
    }
}

int query(int rt,int L,int R,int qL,int qR){
    if(qL <= L && R <= qR) return Tree[rt].sum;
    else{
        int ans = 0;
        if(qL <= mid) ans += query(lc,L,mid,qL,qR);
        if(qR > mid) ans += query(rc,mid+1,R,qL,qR);
        return ans;
    }
}

void PRINT(int rt,int L,int R){
    if(L == R) printf("%d ",Tree[rt].sum);
    else{
        PRINT(lc,L,mid);
        PRINT(rc,mid+1,R);
    }
}

int tot,first[maxn];
void insert(int u,int v){
    tot++;
    e[tot].from = first[u];
    e[tot].v = v;
    first[u] = tot;
}

void dfs(int now,int pre){
    dfn[now] = TIM++;
    tl[dfn[now]] = now;
//    Tree[dfn[now]].sum = 1;
    for(int i = first[now];i;i = e[i].from){
        int v = e[i].v;
        if(v != pre){
            dfs(v,now);
        }
    }dfn_t[now] = TIM++;
    tl[dfn_t[now]] = now; 
}

int main(){
    scanf("%d",&n);
    for(int i = 1;i < n;i++){
        scanf("%d%d",&a,&b);
        insert(a,b);
        insert(b,a);
    }
    
    
    dfs(1,-1);
    TIM--;
//    build(1,1,TIM-1);
    for(int i = 1;i <= n;i++){
        add(1,1,TIM,dfn[i]);
    }
    
    scanf("%d",&m);
    for(int i = 1;i <= m;i++){
        cin >> ctr >> pos;
        if(ctr == 'C') add(1,1,TIM,dfn[pos]);
        else printf("%d
",query(1,1,TIM,dfn[pos],dfn_t[pos]));
//        PRINT(1,1,TIM-1);cout << endl;
    }
    
//    for(int i = 1;i <= n;i++){
//        printf("%d ",dfn[i]);
//    }cout << "##";
    
//    for(int i = 1;i <= TIM;i++) printf("%d ",tl[i]);
    
    return 0;
}