cv1228 苹果树（树链剖分）

题目链接

分析：
练了一下链剖的基本操作

#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

const int N=100005;
int n,m,st[N],tot=0,t[N<<2];
struct node{
    int y,nxt;
};
node way[N<<1];
int top[N],size[N],son[N],deep[N],in[N],out[N],clo=0;

void add(int u,int w)
{
    tot++;
    way[tot].y=w;way[tot].nxt=st[u];st[u]=tot;
    tot++;
    way[tot].y=u;way[tot].nxt=st[w];st[w]=tot;
}

void dfs_1(int now,int fa,int dep)
{
    deep[now]=dep;
    size[now]=1;
    int maxx=0;
    for (int i=st[now];i;i=way[i].nxt)
        if (way[i].y!=fa)
        {
            dfs_1(way[i].y,now,dep+1);
            size[now]+=size[way[i].y];
            if (maxx<size[way[i].y])
                maxx=size[way[i].y],son[now]=way[i].y;
        }
}

void dfs_2(int now,int fa)
{
    if (son[fa]!=now) top[now]=now;
    else top[now]=top[fa];
    in[now]=++clo;
    if (son[now])
    {
        dfs_2(son[now],now);
        for (int i=st[now];i;i=way[i].nxt)
            if (way[i].y!=son[now]&&way[i].y!=fa)
                dfs_2(way[i].y,now);
    }
    out[now]=clo;
}

void update(int bh)
{
    t[bh]=t[bh<<1]+t[bh<<1|1];
}

void build(int bh,int l,int r)
{
    if (l==r)
    {
        t[bh]=1;
        return; 
    }
    int mid=(l+r)>>1;
    build(bh<<1,l,mid);
    build(bh<<1|1,mid+1,r);
    update(bh);
}

void change(int bh,int l,int r,int x)
{
    if (l==r)
    {
        t[bh]^=1;
        return;
    }
    int mid=(l+r)>>1;
    if (x<=mid) change(bh<<1,l,mid,x);
    else change(bh<<1|1,mid+1,r,x);
    update(bh);
}

int ask(int bh,int l,int r,int x,int y)
{
    if (l>=x&&r<=y)
        return t[bh];
    int mid=(l+r)>>1;
    int ans=0;
    if (x<=mid) ans+=ask(bh<<1,l,mid,x,y);
    if (y>mid) ans+=ask(bh<<1|1,mid+1,r,x,y);
    return ans;
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }

    dfs_1(1,0,1);
    dfs_2(1,0);
    build(1,1,n);

    scanf("%d",&m);
    char s[5];
    int x;
    for (int i=1;i<=m;i++)
    {
        scanf("%s%d",&s,&x);
        if (s[0]=='C') change(1,1,n,in[x]);
        else printf("%d\n",ask(1,1,n,in[x],out[x]));
    }
    return 0;
}