L2-012 关于堆的判断 (25 分)

L2-012 关于堆的判断 (25 分)

将一系列给定数字顺序插入一个初始为空的小顶堆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

AC代码

#include<bits/stdc++.h>
#define Inf 0x3f3f3f
const int N = 1005;
using namespace std;
int n, m, cnt, no;
int ans[N];
map<int, int> p;

void create(int x){//堆的建立
	ans[++cnt] = x;
	int t = cnt;
	while(t>1 && ans[t/2]>ans[t]){
		swap(ans[t/2], ans[t]);
		t/=2;
	}
	ans[t] = x;
}
int main(){
	cin >> n >> m;
	for(int i = 1; i<= n; i ++){
		cin >> no;
		create(no);
	}
	for(int i = 1; i <= n; i ++)//记录堆下标
		p[ans[i]] = i;
	while(m--){
		string str;
		int x, y;
		cin >> x >> str;
		if(str[0]=='a'){ //兄弟节点判断 x and y are siblings
			cin >> y >> str >> str;
			if(p[x]/2==p[y]/2)
				cout << "T" << endl;
			else
				cout << "F" << endl;
		}else{
			cin >> str >> str;
			if(str[0] == 'c'){ //子节点判断： x is a child of y
				cin >> str >> y;
				if(p[x]/2==p[y])
					cout << "T" << endl;
				else
					cout << "F" << endl;
			}else if(str[0]=='r'){//根节点判断： x is the root
				if(p[x]==1)
					cout << "T" << endl;
				else
					cout << "F" << endl;
			}else if(str[0]=='p'){//父节点判断： x is the parent of y
				cin >> str >> y;
				if(p[x]==p[y]/2)
				    cout << "T" << endl;
				else
					cout << "F" << endl;
			}
		}
	}
	
	return 0;
}