poj 3321 Apple Tree（树状数组）

链接：http://poj.org/problem?id=3321

题意：一颗有n个分支的苹果树，根为1，每个分支只有一个苹果，给出n-1个分支的关系和给出m个操作，Q x表示询问x的子树（包括x）苹果的数量，C x表示若分支x上有苹果，则摘下来，若没有则会生出一个，输出每个询问的值。

分析：每个分支其实就是一个节点，先dfs整个树，求出每个节点的时间戳，即每个节点第一次访问的时间和最后一次访问的时间，分别用begin和end记录，以时间戳为编号，则在begin[x]和end[x]之间的编号的节点就是x的子树，以时间戳为树状数组的下标，查询时，只要求第一次访问的编号到最后一次访问的编号之间的和就行了，即sum(end[x])-sum(begin[x]-1)；修改时，只要修改第一次访问的编号即可，即update(begin[x])。但是修改前要判断该位置是0或1，0则加1,1则减1。

还有别忘了初始化。

AC代码：

#include<stdio.h>
#include<string.h>
#define N 100010
struct EDGE{
    int v,next;
}edge[N];
int c[N],first[N],begin[N],end[N];
int n,g,count;
void AddEdge(int u,int v)
{
    edge[g].v=v;
    edge[g].next=first[u];
    first[u]=g++;
}
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int num)
{
    while(x<=n)
    {
        c[x]+=num;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int s=0;
    while(x>0)
    {
        s+=c[x];
        x-=lowbit(x);
    }
    return s;
}
void dfs(int u)
{
    int i;
    begin[u]=++count;
    for(i=first[u];i!=-1;i=edge[i].next)
        dfs(edge[i].v);
    end[u]=count;
}
int main()
{
    int u,v,m,ans,i;
    char s[10];
    scanf("%d",&n);
    g=0;
    memset(first,-1,sizeof(first));
    for(i=0;i<n-1;i++)
    {
        scanf("%d%d",&u,&v);
        AddEdge(u,v);
    }
    count=0;
    dfs(1);
    for(i=1;i<=n;i++)
        update(i,1);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        if(s[0]=='Q')
        {
            scanf("%d",&u);
            ans=sum(end[u])-sum(begin[u]-1);
            printf("%d
",ans);
        }
        else
        {
            scanf("%d",&u);
            if(sum(begin[u])-sum(begin[u]-1))
                update(begin[u],-1);
            else
                update(begin[u],1);
        }
    }
    return 0;
}