[poj3321]Apple Tree_dfs序_树状数组

Apple Tree poj-3321

　　　　题目大意：给你一个根固定的树，每一个点的点权是0或1，查询子树点权和。

　　　　注释：$1\le n \le 10^5$。

　　　　　　想法：刚刚学习dfs序，刷到水题偶哈哈。

　　　　　　　　什么是dfs序？就是在遍历树的时候记录的每个点的出栈入栈序。这样就可以保证每一个节会出现两次且它的子树被其夹在中间。

　　　　　　然后，子树信息就可以通过维护序列的鬼东西维护了qwq。

　　　　　　紧接着，我们用树状数组维护被节点夹着的区间，就是端点节点的子树，用树状数组更新即可。

　　　　最后，附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100010
using namespace std;
int tot,cnt;
int to[2*maxn],head[maxn],nxt[2*maxn];
int d[2*maxn];
int p1[maxn],p2[maxn];
int tree[4*maxn];
inline void add(int x,int y)
{
to[++tot]=y;
nxt[tot]=head[x];
head[x]=tot;
}
int lowbit(int x)
{
return x&(-x);
}
void dfs(int pos,int fa)//初始构造dfs序
{
d[++cnt]=1;
p1[pos]=cnt;
for(int i=head[pos];i;i=nxt[i])
{
if(to[i]==fa) continue;
dfs(to[i],pos);
}
p2[pos]=cnt;
}
void fix(int x,int ch)
{
for(int i=x;i<=cnt;i+=lowbit(i))
{
tree[i]+=ch;
}
}
int query(int x)
{
int ans=0;
for(int i=x;i;i-=lowbit(i))
{
ans+=tree[i];
}
return ans;
}
void original()
{
cnt=tot=0;
memset(tree,0,sizeof tree);
memset(head,0,sizeof head);
}
int main()
{
int n,m;
while(~scanf("%d",&n))
{
original();
for(int a,b,i=1;i<n;i++)
{
scanf("%d%d",&a,&b);
add(a,b);
add(b,a);
}
dfs(1,0);
for(int i=1;i<=n;i++)//别忘了建树
{
fix(p1[i],1);
}
// for(int i=1;i<=cnt;i++)
// {
// 	printf("%d ",query(i));
// }
// puts("");
char s[20];
scanf("%d",&m);
for(int x,i=1;i<=m;i++)
{
scanf("%s",s+1);
if(s[1]=='C')
{
scanf("%d",&x);
if(d[p1[x]]==1) fix(p1[x],-1);
else fix(p1[x],1);
d[p1[x]]^=1;
}
else
{
scanf("%d",&x);
printf("%d\n",query(p2[x])-query(p1[x]-1));
}
}
}
return 0;
}
// int main()
// {
// 	int n;
// 	scanf("%d",&n);
// 	for(int a,b,i=1;i<n;i++)
// 	{
// 		scanf("%d%d",&a,&b);
// 		add(a,b);
// 		add(b,a);
// 	}
// 	dfs(1,0);
// 	for(int i=1;i<=cnt;i++)
// 	{
// 		printf("%d ",d[i]);
// 	}
// 	puts("");
// 	for(int i=1;i<=n;i++)
// 	{
// 		printf("%d %d\n",p1[i],p2[i]);
// 	}
// 	return 0;
// }

 　　　　小结：dfs序好东西好东西... ...