poj 3321 apple tree#树状数组#

题目

求一个枝丫上有多少苹果。

---

分析

把树用后序遍历变成树状数组的形式，然后按照信息修改就好了。

---

代码

#include <cstdio>
using namespace std;
struct node{int x,y,next;}e[200000]; bool flag[100001],v[100001];
int ls[100001],n,beg[100001],ed,end[100001],m,c[100001];
void dfs(int x){
	beg[x]=++ed;
	v[x]=1;
	int t=ls[x];
	while (t){
		if (!v[e[t].y]) dfs(e[t].y);
		t=e[t].next;
	}
    end[x]=ed;
}
int lowbit(int x){return x&(-x);}
void add(int x,int y){
	while (x<=m){
		c[x]+=y;
		x+=lowbit(x);
	}
}
int ans(int x){
	int sum=0;
	while (x){
		sum+=c[x];
		x-=lowbit(x);
	}
	return sum;
}
void ad(int i){scanf("%d%d",&e[i].x,&e[i].y); e[i].next=ls[e[i].x]; ls[e[i].x]=i;}
void dd(int i){e[i].x=e[i-m+1].y; e[i].y=e[i-m+1].x; e[i].next=ls[e[i].x]; ls[e[i].x]=i;}
int main(){
	scanf("%d",&m);
	for (int i=1;i<m;i++) ad(i),dd(i+m-1);
	for (int i=1;i<=m;i++) add(i,1);
	dfs(1); scanf("%d\n",&n);//深搜
	while (n--){
		char q=getchar(); int x;
		scanf("%d",&x); if (n) getchar();
		if (q=='C'){
			if (flag[x]) add(beg[x],1); else add(beg[x],-1);//有苹果吗
			flag[x]=!flag[x];
		}
		else printf("%d\n",ans(end[x])-ans(beg[x]-1));
	}
	return 0;
}