pta 5-1 关于堆的判断 (25分)

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;
}