JZOJ 1016. 【PKU3321】苹果树

Description

你家门前种了一棵苹果树，每年秋天，树上都结满了苹果，你非常喜欢吃苹果，所以一直精心照料着苹果树。 苹果树有N个分叉，分叉之间由枝干相连，你把分叉用1到N来标记，树根必须记为1。苹果长在分叉处，而且一个分叉最多只能同时结一个苹果，也就是说不可能有超过一个苹果同时长在分叉处。你想知道某个子树中一共有多少个苹果。 上面的问题不难，但现在的问题是有时你会去摘苹果，有时又长出新的苹果，那你还能做吗？

Input

输入文件第一行是一个整数N(N<=100000),表示树中分叉的数目。 接下来N-1行每行两个整数u和v，表示分叉u和分叉v有一个枝干相连。 下一行是一个整数M(M<=100000) 接下来M行，每行或者是“C x”表示x分叉处的状态发生改变，如果原来有，就被摘下来，原来没有就长出新的苹果；或者是“Q x”表示询问以分叉x为根的子树中苹果的数量。 注意起初每个分叉都有一个苹果。

Output

对于每个询问输出对应的数量。

Sample Input

3
1 2
1 3
3
Q 1
C 2
Q 1

Sample Output

3
2

Sulotion

• 这题是一道典型的线段树，需要单点修改和区间求和。

• 但一眼看去，貌似不知道区间的左右端点

• 但其实在DFS的时候就可以将点处理标号，一个点的子树的标号一定是连续的，

• 那么对于一个点，存住它子树所在的区间，即可进行线段树操作。

Code

#include<cstdio>
using namespace std;
const int N=100001;
int tot;
int first[N],next[N<<1],en[N<<1];
int mx[N],g[N],f[N<<2];
inline int read()
{
    int X=0; char ch=0;
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X;
}
inline void insert(int x,int y)
{
    next[++tot]=first[x];
    first[x]=tot;
    en[tot]=y;
}
inline void dfs(int x,int y)
{
    mx[x]=g[x]=++tot;
    for(int i=first[x];i;i=next[i])
        if(en[i]!=y)
        {
            dfs(en[i],x);
            mx[x]=mx[en[i]];
        }
}
inline void make(int v,int l,int r)
{
    if(l==r)
    {
        f[v]=1;
        return;
    }
    int mid=(l+r)>>1;
    make(v<<1,l,mid);
    make(v<<1|1,mid+1,r);
    f[v]=f[v<<1]+f[v<<1|1];
}
inline void change(int v,int l,int r,int x)
{
    if(l==r)
    {
        f[v]=f[v]^1;
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid) change(v<<1,l,mid,x); else change(v<<1|1,mid+1,r,x);
    f[v]=f[v<<1]+f[v<<1|1];
}
inline int query(int v,int l,int r,int x,int y)
{
    if(l==x && r==y) return f[v];
    int mid=(l+r)>>1;
    if(y<=mid) return query(v<<1,l,mid,x,y);
    if(x>mid) return query(v<<1|1,mid+1,r,x,y);
    return query(v<<1,l,mid,x,mid)+query(v<<1|1,mid+1,r,mid+1,y);
}
int main()
{
    int n=read();
    for(int i=1;i<n;i++)
    {
        int x=read(),y=read();
        insert(x,y);
        insert(y,x);
    }
    dfs(1,tot=0);
    make(1,1,n);
    int m=read();
    while(m--)
    {
        char ch=getchar();
        while(ch!='Q' && ch!='C') ch=getchar();
        int x=read();
        if(ch=='C') change(1,1,n,g[x]); else
            printf("%d\n",query(1,1,n,g[x],mx[x]));
    }
    return 0;
}