L2-012. 关于堆的判断(数据结构)

L2-012. 关于堆的判断

时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越

将一系列给定数字顺序插入一个初始为空的小顶堆H[]。随后判断一系列相关命题是否为真。命题分下列几种:

  • “x is the root”:x是根结点;
  • “x and y are siblings”:x和y是兄弟结点;
  • “x is the parent of y”:x是y的父结点;
  • “x is a child of y”:x是y的一个子结点。

输入格式:

每组测试第1行包含2个正整数N(<= 1000)和M(<= 20),分别是插入元素的个数、以及需要判断的命题数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。之后M行,每行给出一个命题。题目保证命题中的结点键值都是存在的。

输出格式:

对输入的每个命题,如果其为真,则在一行中输出“T”,否则输出“F”。

输入样例:
5 4
46 23 26 24 10
24 is the root
26 and 23 are siblings
46 is the parent of 23
23 is a child of 10
输出样例:
F
T
F
T

提交代码


分析:题目不难~但是有个很大的坑,不能一次性将所有数放入数组之后再更新堆,而要一次次更新~~~

建堆,然后记录每个数的位置(因为数可能为负所以先加上10000)

然后就是结合堆的关系(左儿子是 2*x  右儿子是 2*x+1)得到结果

AC代码:

#include
using namespace std;
int node[200000];
int pre[200000]={0};
void swap(int &a,int &b)
{
	int t;
	t=a;
	a=b;
	b=t;
}
void update(int x,int len)
{
	int t=x;
	if(x*2>len) return;
	if(node[x]>node[x*2])
	t=t*2;
	if(x*2+1<=len&&node[t]>node[x*2+1])
	t=x*2+1;
	if(t==x) return;
	swap(node[x],node[t]);
	update(t,len);
}
void buildtree(int len)
{
	for(int i=len/2;i>=1;i--)
	{
		update(i,len);
	}
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	memset(node,-1,sizeof(node));
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&node[i]);
		node[i]+=10000;
		buildtree(i);
	}
	//buildtree(n);
	for(int i=1;i<=n;i++)
	pre[node[i]]=i;
	while(m--)
	{
		int a,c;
		scanf("%d",&a);
		a+=10000;
		char b[100];
		scanf("%s",b);
		if(b[0]=='a')
		{
			scanf("%d",&c);
			c+=10000;
			scanf("%s%s",b,b);
			if(pre[a]/2==pre[c]/2&&pre[a]!=pre[c])
			printf("T\n");
			else
			printf("F\n");
		}
		else
		{
			scanf("%s",b);
			if(b[0]=='a')
			{
				scanf("%s%s",b,b);
				scanf("%d",&c);
				c+=10000;
				if(pre[c]*2==pre[a]||pre[c]*2+1==pre[a])
				printf("T\n");
				else
				printf("F\n");
			}
			else
			{
				scanf("%s",b);
				if(b[0]=='r')
				{
					if(pre[a]==1) 
					printf("T\n");
					else
					printf("F\n");
				}
				else
				{
					scanf("%s",b);
					scanf("%d",&c);
					c+=10000;
					if(pre[a]==pre[c]/2)
					printf("T\n");
					else
					printf("F\n");
				}
			}
		}
	}
}



你可能感兴趣的:(团体程序设计天梯赛(PAT))