AC日记——苹果树 codevs 1228

时间:2021-01-23 17:44:51

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

思路:

  首先这个题是一棵树

  他的点的编号顺序并不是按照线段树的节点顺序来排列的

  所以我们要想方设法把这个节点融入线段树中

  如何融入线段树呢?

  首先把这些点和边都存起来

  然后我们考虑把节点存成一个线性的区间

  定义一个cnt来记录当前的节点的编号和它所连的点(不包括他被延伸的节点)的

  然后从第一个节点开始遍历

  用dfs序来延伸,并把新点不断存入区间

  然后所有的节点所能延伸的点包括自己都能被表示为一个区间

  然后每个节点就是就是它的区间的第一个值

  然后线段树完成单点修改和区间查询

  轻松水过

来,上代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm> using namespace std; struct T_tree {
int l,r,dis;
};
struct T_tree tree[]; struct T_edge {
int from,to,next;
};
struct T_edge edge[]; struct T_interval {
int li,ri;
};
struct T_interval find_[]; int n,m,num_edge=,head[],tot=; 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<<].dis+tree[now<<|].dis;
} void tree_build(int now,int l,int r)
{
tree[now].l=l,tree[now].r=r;
if(l==r)
{
tree[now].dis=;
return ;
}
int mid=(l+r)>>;
tree_build(now<<,l,mid);
tree_build(now<<|,mid+,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=;
else tree[now].dis=;
return ;
}
int mid=(tree[now].l+tree[now].r)>>;
if(to>mid) tree_change(now<<|,to);
else tree_change(now<<,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)>>;
if(l>mid) return tree_query(now<<|,l,r);
else if(r<=mid) return tree_query(now<<,l,r);
else return tree_query(now<<,l,mid)+tree_query(now<<|,mid+,r);
} int main()
{
scanf("%d",&n);
int from,to;
for(int i=;i<n;i++)
{
scanf("%d%d",&from,&to);
edge_add(from,to);
}
dfs();
tree_build(,,n);
scanf("%d",&m);
char t_do;
int to_ch;
for(int i=;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\n",tree_query(,find_[to_ch].li,find_[to_ch].ri));
else tree_change(,find_[to_ch].li);
}
return ;
}