关闭

poj 3321 Apple Tree (树状数组+DFS)

标签: 树状数组前向星
287人阅读 评论(0) 收藏 举报
分类:

题目:http://poj.org/problem?id=3321

题意:给定一个整数n,代表树的节点个数,在给出n-1条边(u,v),其中编号1是根节点,开始时每个节点的值都为1。有两种操作①C X 将节点x的值取非(1变0,0变1),②Q x查询x的子树(包括x)值的总和。

分析:题目说的(u,v),只告诉u和v是相连的,但我当有向边做的也AC了。。。无向边怎么做???树上的边都是有向的吧?

首先要把树建起来。建好后,修改节点的值好办,直接用树状数组log(n)修改,查询就不好办了,因为一个节点它的子树的编号可能是无序的,查询起来不方便。那么建树的时候把所有的节点重新编号就行了。。。查询的子树就是一段连续的区间了。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn = 1e5+5;

int trans[maxn],L[maxn],R[maxn];
bool f[maxn];

struct node
{
	int pos;
	int next;
};

int Table[maxn],cnt;
node List[maxn];
void Clear()
{
	cnt=0;
	memset(Table,-1,sizeof(Table)); 
}
void Insert(int Tbl,int v)
{
	List[cnt].pos=v;
	List[cnt].next=Table[Tbl];
	Table[Tbl]=cnt++;
}

int cur,tree[maxn];

void DFS(int value)
{
	trans[value]=++cur;
	L[trans[value]]=cur;
	int next=Table[value],i;
	while(next>=0)
	{
		DFS(List[next].pos);
		next=List[next].next;
	}
	R[trans[value]]=cur;
}

int lowbit(int x)
{
	return x&-x;
}

void update(int x,int c)
{
	for(int i=x;i<=cur && i;i+=lowbit(i))
		tree[i]+=c;
}

int query(int x)
{
	int ret(0);
	for(int i=x;i>0;i-=lowbit(i))
		ret+=tree[i];
	return ret;
}

int main()
{
	int n,u,v,m,c;
	char str[20];
	while(scanf("%d",&n)!=EOF)
	{
		Clear(); 
		memset(tree,0,sizeof(tree));
		
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&u,&v);
			Insert(u,v);
		}
		cur=0;
		DFS(1);

		memset(f,true,sizeof(f));
		for(int i=1;i<=cur;i++)
			update(i,1);

		scanf("%d",&m);
		while(m--)
		{
			scanf("%s%d",str,&c);
			if(str[0]=='Q')
				printf("%d\n",query(R[trans[c]])-query(L[trans[c]]-1));
			else
			{
				if(f[trans[c]])
				{
					update(trans[c],-1);
					f[trans[c]]=false;
				}
				else
				{
					update(trans[c],1);
					f[trans[c]]=true;
				}
			}
		}
	}
	return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:73465次
    • 积分:2684
    • 等级:
    • 排名:第14012名
    • 原创:201篇
    • 转载:2篇
    • 译文:0篇
    • 评论:5条
    最新评论