5-1 关于堆的判断 (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
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <functional>
using namespace std;
vector<int> v;
int Find(int a) {//查找对应元素值的下标
return find(v.begin()+1, v.end(), a)-v.begin();
}
int main(int argc, char const *argv[]) {
int n, m, a, b;
v.push_back(0);//0坐标用掉,这样比较方便
//父亲结点就是i / 2 儿子结点2 * i or 2 * i + 1
char s[100], tmp[4];
scanf("%d %d", &n, &m);
for(int i = 0; i < n; ++i) {
scanf("%d", &a);
v.push_back(a);
push_heap(v.begin()+1, v.end(), greater<int>());//对加入的元素构造最小堆
}
scanf("%*c");
while(m--) {
gets(s);
if(s[strlen(s) - 1] == 't') {
sscanf(s, "%d", &a);//提取字符串里面的数字
if(v[1] == a) printf("T\n");//如果是根结点
else printf("F\n");
}
else if(s[strlen(s) - 1] == 's') {
sscanf(s, "%d %*s %d", &a, &b);
a = Find(a);
b = Find(b);
if(a/2 == b/2) printf("T\n");//兄弟结点
else printf("F\n");
}
else {
sscanf(s, "%d %*s %s %*s %*s %d", &a, tmp, &b);
a = Find(a);
b = Find(b);
if(!strcmp(tmp, "the")) {
if(a == b/2) printf("T\n");//a是b的父亲结点
else printf("F\n");
}
else {
if(a/2 == b) printf("T\n");//a是b的儿子结点
else printf("F\n");
}
}
}
return 0;
}