Codevs 苹果树

点击就送屠龙宝刀

题目描述 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;}