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

最新推荐文章于 2021-04-11 10:22:26 发布

原创最新推荐文章于 2021-04-11 10:22:26 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#堆

C++ STL 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用C++ STL实现小顶堆的方法，并通过具体实例演示了如何进行结点关系查询，包括判断根节点、兄弟节点、父子节点等关系。通过对代码的解析，读者可以了解到小顶堆的基本操作及其实现细节。

将一系列给定数字顺序插入一个初始为空的小顶堆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

输出样例：

没有手写过堆不会写...也可以用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;
}