5-12 关于堆的判断 (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

（\le≤ 1000）和

M

（\le≤ 20），分别是插入元素的个数、以及需要判断的命题数。下一行给出区间[-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

没有手写过堆 不会写...也可以用STL的heap来实现， 还是heap实现起来方便多了。

#include<bits/stdc++.h>
using namespace std;

vector<int>heap;
int n, m;

int find(int a)
{
return distance(heap.begin(), find(heap.begin(), heap.end(), a));   //找a到顶的距离，也就是他的对应下标
}

int main(void )
{
scanf("%d%d", &n, &m);
while(n--)
{
int t;
scanf("%d", &t);
heap.push_back(t);
make_heap(heap.begin(), heap.end(), greater<int>());    //小顶堆
}
getchar();
while(m--)
{
string q;
getline(cin, q);
if(q[q.length() - 1] == 't')
{
int a;
stringstream ss(q);
ss >> a;
if (heap[0] == a) printf("T\n");
else printf("F\n");
}
else if(q[q.length() - 1] == 's')
{
int a, b;
string temp;
stringstream ss(q);
ss >> a >> temp >> b;
int pos_a = find(a);
int pos_b = find(b);    //根是0，所以要加1
pos_a++; pos_b++;
if(pos_a / 2 == pos_b / 2) // /2是找父亲，如果父亲相同就是兄弟
printf("T\n");
else
printf("F\n");
}
else
{
stringstream ss(q);
int a, b;
string temp;
ss >> a >> temp >> temp;
if(temp[0] == 't')
{
ss >> temp >> temp >> b;
int pos_a = find(a);
int pos_b = find(b);
if(pos_b == pos_a * 2 + 1 || pos_b == pos_a * 2 + 2)
printf("T\n");
else
printf("F\n");
}
else if(temp[0] == 'a')
{
ss >> temp >> temp >> b;
int pos_a = find(a);
int pos_b = find(b);
if(pos_a == pos_b * 2 + 1 || pos_a == pos_b * 2 + 2)
printf("T\n");
else
printf("F\n");
}
}
}
return 0;
}