天梯赛 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
)
(≤ 1000)
(≤1000)和M
(
≤
20
)
(≤ 20)
(≤20),分别是插入元素的个数、以及需要判断的命题数。下一行给出区间
[
−
10000
,
10000
]
[−10000,10000]
[−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
简单分析:
就是一个手动模拟堆的过程,调用priority_queue是解决不了的= =
建堆的时候是执行up操作,不能用down操作。
还要记录一下下标的位置,使用集合存储,方便查询时候输出。
就是查询的时候字符串处理比较麻烦。
代码
#include <iostream>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N = 10010;
int n, m;
int h[N], cnt;
void up(int u) {
while (u >> 1 && h[u] < h[u >> 1]) {
swap(h[u], h[u >> 1]);
u >>= 1;
}
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n; i++) {
int x;
cin >> x;
h[++cnt] = x;
up(i);
}
unordered_map<int, int> mp;
for (int i = 1; i <= n; i++) mp[h[i]] = i;
while (m--) {
int a;
cin >> a;
string s1;
cin >> s1;
if (s1 == "and") {
int b;
cin >> b;
cin >> s1 >> s1;
if (mp[a] / 2 == mp[b] / 2) puts("T");
else puts("F");
}
else {
cin >> s1;
if (s1 == "a") {
cin >> s1 >> s1;
int b;
cin >> b;
if (mp[a] / 2 == mp[b]) puts("T");
else puts("F");
}
else {
cin >> s1;
if (s1 == "root") {
if (a == h[1]) puts("T");
else puts("F");
}
else {
cin >> s1;
int b;
cin >> b;
if (mp[b] / 2 == mp[a]) puts("T");
else puts("F");
}
}
}
}
return 0;
}
感悟与收获
这里建堆操作只能使用up。