题目描述 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
思路:
首先这个题是一棵树
他的点的编号顺序并不是按照线段树的节点顺序来排列的
所以我们要想方设法把这个节点融入线段树中
如何融入线段树呢?
首先把这些点和边都存起来
然后我们考虑把节点存成一个线性的区间
定义一个cnt来记录当前的节点的编号和它所连的点(不包括他被延伸的节点)的
然后从第一个节点开始遍历
用dfs序来延伸,并把新点不断存入区间
然后所有的节点所能延伸的点包括自己都能被表示为一个区间
然后每个节点就是就是它的区间的第一个值
然后线段树完成单点修改和区间查询
轻松水过
来,上代码:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; struct T_tree { int l,r,dis; }; struct T_tree tree[400001]; struct T_edge { int from,to,next; }; struct T_edge edge[100001]; struct T_interval { int li,ri; }; struct T_interval find_[100001]; int n,m,num_edge=0,head[100001],tot=0; inline void edge_add(int from,int to) { num_edge++; edge[num_edge].to=to; edge[num_edge].from=from; edge[num_edge].next=head[from]; head[from]=num_edge; } void dfs(int now) { tot++; find_[now].li=tot; for(int i=head[now];i;i=edge[i].next) { dfs(edge[i].to); } find_[now].ri=tot; } void tree_up(int now) { tree[now].dis=tree[now<<1].dis+tree[now<<1|1].dis; } void tree_build(int now,int l,int r) { tree[now].l=l,tree[now].r=r; if(l==r) { tree[now].dis=1; return ; } int mid=(l+r)>>1; tree_build(now<<1,l,mid); tree_build(now<<1|1,mid+1,r); tree_up(now); return ; } void tree_change(int now,int to) { if(to==tree[now].l&&tree[now].r==to) { if(tree[now].dis) tree[now].dis=0; else tree[now].dis=1; return ; } int mid=(tree[now].l+tree[now].r)>>1; if(to>mid) tree_change(now<<1|1,to); else tree_change(now<<1,to); tree_up(now); } int tree_query(int now,int l,int r) { if(tree[now].l==l&&tree[now].r==r) { return tree[now].dis; } int mid=(tree[now].l+tree[now].r)>>1; if(l>mid) return tree_query(now<<1|1,l,r); else if(r<=mid) return tree_query(now<<1,l,r); else return tree_query(now<<1,l,mid)+tree_query(now<<1|1,mid+1,r); } int main() { scanf("%d",&n); int from,to; for(int i=1;i<n;i++) { scanf("%d%d",&from,&to); edge_add(from,to); } dfs(1); tree_build(1,1,n); scanf("%d",&m); char t_do; int to_ch; for(int i=1;i<=m;i++) { //scanf("%c %d",&t_do,&to_ch); //scanf("%c",&t_do); //scanf("%d",&to_ch); cin>>t_do>>to_ch; if(t_do=='Q') printf("%d ",tree_query(1,find_[to_ch].li,find_[to_ch].ri)); else tree_change(1,find_[to_ch].li); } return 0; }