poj 3321 Apple Tree DFS时间戳预处理+树状数组

分析:本题的操作乍一看很像树状数组,单点更新,查询连续子序列和。多完美,可仔细一想不对,如何用一种序列来表示枝叉呢,这是一个麻烦的问题。换个角度想想,如果表示一个树枝上所有节点呢。答案是有的,dfs。从根开始深搜这棵树,访问每个点并加上时间戳。那么一个树枝上的所有节点就可以用时间表示,就是访问树枝的根的时间和后的时间之间。但是要注意的是,每个点都可以用两个时间表示,dfs前的时间和dfs后的时间。那么对某个点求果子数目就很简单了,直接用树状数组标准求前缀和的思想,然后求差,再除以2就是答案。详见代码。

代码:

#include<cstdio>
#include<cstring>
//5440K	407MS
const int maxn  = 1e5 + 5;
int c[maxn<<1];
int n;
int clock=0;
bool fork[maxn];
int pre[maxn];
int post[maxn];
int pool;
struct Edge{
	int to;
	int next;	
}edge[maxn<<1];
int head[maxn];
bool visit[maxn];
inline int lowbit(int x){
	return x&-x;
}
void add(int x,int a){
	while(x<=n){
		c[x]+=a;
		x+=lowbit(x);
	}
}
int sum(int x){
	int ret=0;
	while(x>0){
		ret+=c[x];
		x-=lowbit(x);
	}
	return ret;
}
void buildtree()
{
	for(int i=1;i<=n;i++){
		c[i]=lowbit(i);
	}
}
inline void addedge(int from,int to){
	edge[++pool].to=to;
	edge[pool].next=head[from];
	head[from]=pool;
}
void dfs(int i){
	pre[i]=++clock;
	visit[i]=true;
	for(int k=head[i];k!=-1;k=edge[k].next){
		if(!visit[edge[k].to]){
			dfs(edge[k].to);
		}
	}
	post[i]=++clock;
}
int main()
{
	int u,v;
	memset(head,-1,sizeof(head));
	scanf("%d",&n);
	n<<=1;
	buildtree();
	int k=n/2-1;
	for(int i=0;i<k;i++){
		scanf("%d %d",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	dfs(1);
	int m,x,z;char str[2];
	scanf("%d",&m);
	while(m--){
		scanf("%s %d",str,&x);
		if(str[0]=='C'){
			fork[x]=!fork[x];
			if(fork[x]) z=-1;
			else z=1;
			add(pre[x],z);
			add(post[x],z);
		}else{
			printf("%d\n",(sum(post[x])-sum(pre[x]-1))>>1);
		}
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值