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

原创 2015年07月09日 14:36:00

题目: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;
}

版权声明:转吧。

相关文章推荐

【树状数组--dfs序转化】poj3321 Apple Tree

Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 29822   Accepted: 8881...

poj 3321 Apple Tree(dfs染色+树状数组)

Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 27212   Ac...

【poj3321】Apple Tree——树状数组&DFS

【poj3321】Apple Tree——树状数组&DFS##include #include #include using namespace std; const int maxn = 10...
  • CFhM_R
  • CFhM_R
  • 2016年05月29日 14:24
  • 150

Apple Tree_poj3321_树状数组&dfs

Description There is an apple tree outside of kaka'shouse. Every autumn, a lot of apples will grow i...
  • jpwang8
  • jpwang8
  • 2016年05月23日 20:54
  • 286

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

Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 16777   Ac...
  • WEYuLi
  • WEYuLi
  • 2013年11月27日 20:37
  • 982

POJ 3321 APPLE TREE(树状数组+DFS)

原文:http://blog.acmj1991.com/?p=669 题意:有一颗苹果树根为1,给出你每个叉点之间的关系。初始状态时每个叉点上有一个苹果。接下来有两种操作:C x 改变叉点x的状...

POJ 3321-Apple Tree(DFS序+树状数组)

题目链接:http://poj.org/problem?id=3321 大致题意:给你一棵树,有n个点,每个点初始时权值为1,有两种操作: (1)C x :若节点x的权值为1,则将其变为0...

[poj 3321]:Apple Tree(树状数组/线段树 和dfs序)

Apple Tree Time Limit: 2000MS Memory Limit: 65536K Total Submissions: 29084 Accepted: ...
  • xljer_
  • xljer_
  • 2017年05月24日 14:08
  • 504

【poj 3321】 Apple Tree 树状数组+dfs序

Apple Tree Time Limit: 2000MS Memory Limit: 65536K Total Submissions: 23568 Accepted: ...
  • ALPS233
  • ALPS233
  • 2016年04月05日 15:05
  • 668

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

Apple Tree Description There is an apple tree outside of kaka's house. Every autumn, a lot of ap...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:poj 3321 Apple Tree (树状数组+DFS)
举报原因:
原因补充:

(最多只允许输入30个字)