POJ_3321 Apple Tree

题意

给出一个苹果树的一些分叉(x,y),表示y是x的子节点,给出m次操作,有两种:
1、C x 如果叉x有苹果就摘,没有就长,相当于取反。
2、Q x 查询叉x(包括x)的子节点上一共有多少苹果。

思路

先进行深度优先遍历求出这棵树的先序后序遍历顺序,然后我们进行C操作时就判断然后取反,进行Q操作时就用find(end[x])-find(begin[x]),begin是先序遍历,end是后序遍历,这样子我们可以求出x的子节点的苹果数量了。

代码

#include<cstdio>
struct node{
    int next,to;
}edge[100001];
int tot,n,m,x,tree[100001],head[100001],begin[100001],end[100001],apple[100001];
int c;
int lowbit(int x) {return x&-x;}
void add(int x,int v)
{
    for (;x<=n;x+=lowbit(x))
        tree[x]+=v;
}
int find(int x)
{
    int o=0;
    for (;x>0;x-=lowbit(x))
        o+=tree[x];
    return o;
}
void dfs(int x)
{
    begin[x]=tot;
    for (int i=head[x];i;i=edge[i].next) dfs(edge[i].to);
    end[x]=++tot;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&edge[i].to);//邻接表优化遍历
        edge[i].next=head[x];
        head[x]=i;
        add(i,1);//边输入边初始化
    }
    add(n,1);
    dfs(1);//求先序后序遍历
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        scanf("\n%c %d",&c,&x);
        if (c=='C')
        {
            apple[x]^=1;//取反
            if (apple[x]) add(end[x],-1);//因为有苹果所以摘
            else add(end[x],1);//相反
        }
        else printf("%d\n",find(end[x])-find(begin[x]));
    }
}

猜你喜欢

转载自blog.csdn.net/ssl_hzb/article/details/80292823