L2-012 關於堆的判斷

                                                     L2-012 關於堆的判斷

                           https://pintia.cn/problem-sets/994805046380707840/problems/994805064676261888

 

 

題目

將一系列給定數字順序插入一個初始為空的小頂堆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

分析

二叉堆

C++程式

#include<iostream>
#include<string>
#include<map>

using namespace std;

const int N=1005;

int a[N];
map<int,int>pos;

//向堆中插入元素 
void up(int start)
{
	int c=start,tmp=a[c];
	while(c>0&&a[(c-1)/2]>tmp)
	{
		a[c]=a[(c-1)/2];
		c=(c-1)/2;
	}
	a[c]=tmp;
}

int main()
{
	int n,m,x,y;
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
		up(i);
	}
	for(int i=0;i<n;i++)
	  pos[a[i]]=i;//元素a[i]的位置在i 
	while(m--)
	{
		string s,s1,s2,s3,tmp;
		cin>>x>>s;
		if(s=="is")
		{
			cin>>s1>>s2;
			s3=s1+s2;
			if(s3=="theroot")
			{
				printf("%c\n",(pos[x]==0)?'T':'F');
			}
			else if(s3=="theparent")
			{
				cin>>tmp>>y;
				printf("%c\n",((pos[y]-1)/2==pos[x])?'T':'F');
			}
			else if(s3=="achild")
			{
				cin>>tmp>>y;
				printf("%c\n",((pos[x]-1)/2==pos[y])?'T':'F');
			}
		}
		else 
		{
			cin>>y;
			getline(cin,tmp);
			//是否為兄弟
			printf("%c\n",((pos[x]-1)/2==(pos[y]-1)/2)?'T':'F');
		}
	}
	return 0;
}