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;
}