codevs——1228 苹果树

Posted 2020-10-07

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

 

拿到这个题目以后，我们第一眼看到的会是：有m次询问，每次询问2种操作。那么就可以很快的get到这个题是要用线段树。

但是问题又来了，现在他给我们的是一个树啊，而且我们线段树维护的是一个序列，那我们就要先办法将这棵树转化成一个序列

我们想一下要用什么方法完成这个操作呢??  对！就是dfs序！

我们现对于我们建出来的树跑一遍dfs，然后求出每一个节点的dfs序，同时处理出每一个节点的子树的大小，这样我们可以保证每一棵子树内的所有的节点都是相连的。以一个节点为子树的区间即为从这个节点到这个节点+其子树的大小+1，当我们询问一个节点的苹果个数是既可以转换成询问这个区间的大小。在我们进行苹果的更改时，即为对一个点进行单点修改，我们进行单点修改的时候我们直接对其进行取反既可以。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 110000
using namespace std;
char ch;
int n,m,q,x,y,s,tot,ans;
int head[N],size[N],list1[N],list2[N];
int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1; ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘; ch=getchar();}
    return x*f;
}
struct Edge
{
    int next,to,from;
}edge[N<<1];
int add(int x,int y)
{
    tot++;
    edge[tot].to=y;
    edge[tot].next=head[x];
    head[x]=tot;
}
struct Tree
{
    int l,r,w;
}tree[N*4];
int dfs(int x)
{
    size[x]=1,list1[x]=++s,list2[s]=x;
    for(int i=head[x];i;i=edge[i].next)
    {
        int t=edge[i].to;
        dfs(t);
        size[x]+=size[t];
    }
}
void build(int k,int l,int r)
{
    tree[k].l=l,tree[k].r=r;
    if(tree[k].l==tree[k].r)
    {
        tree[k].w=1;
        return ;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    tree[k].w=tree[k<<1].w+tree[k<<1|1].w;
}
void change_point(int k)
{
    if(tree[k].l==tree[k].r)
    {
        tree[k].w=!tree[k].w;
        return ;
    }
    int mid=(tree[k].r+tree[k].l)/2;
    if(x<=mid) change_point(k<<1);
    else change_point(k<<1|1);
    tree[k].w=tree[k<<1].w+tree[k<<1|1].w;
}
void ask_interval(int k)
{
    if(tree[k].l>=x&&tree[k].r<=y)
    {
        ans+=tree[k].w;
        return ;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(x<=mid) ask_interval(k<<1);
    if(y>mid)  ask_interval(k<<1|1);
}
int main()
{
    n=read();
    for(int i=1;i<n;i++)
     x=read(),y=read(),add(x,y);
    dfs(1);
    build(1,1,n);
    m=read();
    while(m--)
    {
        cin>>ch;
        q=read();
        x=list1[q];ans=0;
        if(ch==‘C‘) change_point(1);
        else 
        {
            y=x+size[q]-1;ask_interval(1);
            printf("%d\n",ans);
        }
    }
    return 0;
}