【树的遍历 树状数组】POJ_3321 Apple Tree

本文主要是介绍【树的遍历 树状数组】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]));}
}

这篇关于【树的遍历 树状数组】POJ_3321 Apple Tree的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---