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]$内的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;
}