题目描述 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
听说大神们都用链剖做这道题。。我这种今天刚学链剖的蒟蒻就不打啦我怕无限WA。。。于是潜(zi)心(xi)研(yue)究(du)题(ti)目(jie),用线段树A了这道题。。
首先我们用dfs求出对于每一个点可以连接到多少个“儿子”。为什么儿子要有引号呢,因为在这里儿子不一定是儿子还可能是父亲节点。。但是我们可以假定有若干个儿子,用一个数组g来记录儿子的数目。然后我们就得到了一个记录着某个点连接多少个节点的数组。因为刚开始每个分叉都有一个苹果,那么对于每一个点,它的苹果数就是所有儿子节点的加上自己的。那这样就简单了。利用g数组构建一颗线段树,然后进行单点修改和单点查询。注意单点查询的节点x并不是tree[x].sum而是g[x]修改后的值。所以这个并不是一个线段树裸题,我们只是用线段树来维护查询和修改从而使程序的复杂度降低到nlogn的级别。纯粹是优化。当然树状数组也是可以的嘛。。
简单说下链剖的做法:开始时点权值全部为1,然后维护重边和父亲节点。但是要注意,用链剖做的话查询一个节点x直接返回这个节点x的父亲f[x]的值然后减去深度两者的深度差+1就好了。
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int l[300008],r[300008],d[300008],head[300008];
bool si[300008];
int tt=0,n,m;
struct NODE
{
int pre;
int re;
int next;
} a[600014];
struct Node
{
int left;
int right;
int sum;
int la;
} x[300008];
void dfs(int k)
{
l[k]=++tt;
int to;
int i;
for(i=head[k];i;i=a[i].next)
{
to=a[i].re;
dfs(to);
}
r[k]=tt;
}
void build(int l,int r,int s)
{
x[s].left=l;
x[s].right=r;
x[s].sum=0;
x[s].la=0;
if(l==r)
{
x[s].sum=1;
return ;
}
int mid=(l+r)>>1;
build(l,mid,s<<1);
build(mid+1,r,s<<1|1);
x[s].sum=x[s<<1].sum+x[s<<1|1].sum;
}
void change(int l,int r,int val,int s)
{
if(x[s].left==l&&x[s].right==r)
{
x[s].sum+=(x[s].right-x[s].left+1)*val;
x[s].la+=val;
return ;
}
if(x[s].la)
{
x[s<<1].sum+=(x[s<<1].right-x[s<<1].left+1)*x[s].la;
x[s<<1|1].sum+=(x[s<<1|1].right-x[s<<1|1].left+1)*x[s].la;
x[s<<1].la+=x[s].la;
x[s<<1|1].la+=x[s].la;
x[s].la=0;
}
int mid=(x[s].left+x[s].right)>>1;
if(l>mid)change(l,r,val,s<<1|1);
if(r<=mid)change(l,r,val,s<<1);
if(l<=mid&&r>mid)change(l,mid,val,s<<1),change(mid+1,r,val,s<<1|1);
x[s].sum=x[s<<1].sum+x[s<<1|1].sum;
}
int ans(int l,int r,int s)
{
if(x[s].left==l&&x[s].right==r)
return x[s].sum;
if(x[s].la)
{
x[s<<1].sum+=(x[s<<1].right-x[s<<1].left+1)*x[s].la;
x[s<<1|1].sum+=(x[s<<1|1].right-x[s<<1|1].left+1)*x[s].la;
x[s<<1].la+=x[s].la;
x[s<<1|1].la+=x[s].la;
x[s].la=0;
}
int mid=(x[s].left+x[s].right)>>1;
if(l>mid)return ans(l,r,s<<1|1);
if(r<=mid)return ans(l,r,s<<1);
if(l<=mid&&r>mid)return ans(l,mid,s<<1)+ans(mid+1,r,s<<1|1);
}
int main()
{
scanf("%d",&n);
int i;
int b,c;
int num=0;
for(i=1;i<n;++i)
{
scanf("%d%d",&b,&c);
num++;
a[num].pre=b;
a[num].re=c;
a[num].next=head[b];
head[b]=num;
}
dfs(1);
build(1,n,1);
char temp[2];
scanf("%d",&m);
int g;
for(i=1;i<=m;++i)
{
cin>>temp;
if(temp[0]=='C')
{
scanf("%d",&c);
if(d[c]==0)
{
d[c]^=1;
change(l[c],l[c],-1,1);
}
else
{
d[c]^=1;
change(l[c],l[c],1,1);
}
}
else
{
scanf("%d",&g);
printf("%d\n",ans(l[g],r[g],1));
}
}
return 0;
}