[Codevs] 1228 苹果树

Posted 2020-10-06 Lev今天学习了吗

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

  1 #include<cstdio>
  2 #include<iostream>
  3 #define mid (L+R)/2
  4 #define lc (rt<<1)
  5 #define rc (rt<<1|1)
  6 #define maxn 1000000
  7 using namespace std;
  8 
  9 int dfn[maxn],TIM = 1,dfn_t[maxn],n,a,b,c,m,pos,tl[maxn];
 10 char ctr;
 11 
 12 struct edge{
 13     int from,v;
 14 }e[maxn];
 15 
 16 struct node{
 17     int sum;
 18 }Tree[maxn];
 19 
 20 void maintain(int rt){
 21     Tree[rt].sum = Tree[lc].sum+Tree[rc].sum;
 22 }
 23 
 24 void build(int rt,int L,int R){
 25     if(L == R) Tree[rt].sum = 1;
 26     else{
 27         build(lc,L,mid);
 28         build(rc,mid+1,R);
 29         
 30         maintain(rt);
 31     }
 32 }
 33 
 34 void add(int rt,int L,int R,int pos){
 35     if(L == R) Tree[rt].sum ^= 1;
 36     else{
 37         if(pos <= mid) add(lc,L,mid,pos);
 38         else add(rc,mid+1,R,pos);
 39         
 40         maintain(rt);
 41     }
 42 }
 43 
 44 int query(int rt,int L,int R,int qL,int qR){
 45     if(qL <= L && R <= qR) return Tree[rt].sum;
 46     else{
 47         int ans = 0;
 48         if(qL <= mid) ans += query(lc,L,mid,qL,qR);
 49         if(qR > mid) ans += query(rc,mid+1,R,qL,qR);
 50         return ans;
 51     }
 52 }
 53 
 54 void PRINT(int rt,int L,int R){
 55     if(L == R) printf("%d ",Tree[rt].sum);
 56     else{
 57         PRINT(lc,L,mid);
 58         PRINT(rc,mid+1,R);
 59     }
 60 }
 61 
 62 int tot,first[maxn];
 63 void insert(int u,int v){
 64     tot++;
 65     e[tot].from = first[u];
 66     e[tot].v = v;
 67     first[u] = tot;
 68 }
 69 
 70 void dfs(int now,int pre){
 71     dfn[now] = TIM++;
 72     tl[dfn[now]] = now;
 73 //    Tree[dfn[now]].sum = 1;
 74     for(int i = first[now];i;i = e[i].from){
 75         int v = e[i].v;
 76         if(v != pre){
 77             dfs(v,now);
 78         }
 79     }dfn_t[now] = TIM++;
 80     tl[dfn_t[now]] = now; 
 81 }
 82 
 83 int main(){
 84     scanf("%d",&n);
 85     for(int i = 1;i < n;i++){
 86         scanf("%d%d",&a,&b);
 87         insert(a,b);
 88         insert(b,a);
 89     }
 90     
 91     
 92     dfs(1,-1);
 93     TIM--;
 94 //    build(1,1,TIM-1);
 95     for(int i = 1;i <= n;i++){
 96         add(1,1,TIM,dfn[i]);
 97     }
 98     
 99     scanf("%d",&m);
100     for(int i = 1;i <= m;i++){
101         cin >> ctr >> pos;
102         if(ctr == ‘C‘) add(1,1,TIM,dfn[pos]);
103         else printf("%d\n",query(1,1,TIM,dfn[pos],dfn_t[pos]));
104 //        PRINT(1,1,TIM-1);cout << endl;
105     }
106     
107 //    for(int i = 1;i <= n;i++){
108 //        printf("%d ",dfn[i]);
109 //    }cout << "##";
110     
111 //    for(int i = 1;i <= TIM;i++) printf("%d ",tl[i]);
112     
113     return 0;
114 }

情绪低落= =