PAT 天梯赛 L2-012. 关于堆的判断 【Heap】【STL】

题目链接

https://www.patest.cn/contests/gplt/L2-012

思路

使用 STL 里面有关 Heap 的函数

std::make_heap将[start, end)范围进行堆排序，默认使用less, 即最大元素放在第一个。

std::pop_heap将front（即第一个最大元素）移动到end的前部，同时将剩下的元素重新构造成(堆排序)一个新的heap。

std::push_heap对刚插入的（尾部）元素做堆排序。

std::sort_heap将一个堆做排序,最终成为一个有序的系列，可以看到sort_heap时，必须先是一个堆（两个特性：1、最大元素在第一个 2、添加或者删除元素以对数时间），因此必须先做一次make_heap.

make_heap, pop_heap, push_heap, sort_heap都是标准算法库里的模板函数，用于将存储在vector/deque 中的元素进行堆操作

根据 题目的意思 是要用 push_heap 一个一个 插入

而不能用 make_heap

因为 push_heap 和 make_heap 构造的堆 是不一样的

比如

3 6 2 4 1 5 这组数据

make_heap

push_heap

然后 用一个 map 将元素 与其 数组下标 对应起来 要从1 开始

根节点 就是判断一下 其数组下标 是不是 1

x 和 y 是兄弟节点

就是 判断一下

a = x 的数组下标 / 2

b = y 的数组下标 / 2

a 对应的数 和b 对应的数 是不是相等 并且 a != b

x 是 y 的父节点

就是 判断一下 y 的数组下标 / 2 是不是 等于 x 的数组下标

x 是

x 是 y 的子节点

就是 判断一下 x 的数组下标 / 2 是不是 等于 y 的数组下标

AC 代码

#include <cstdio>
#include <cstring>
#include <ctype.h>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <map>
#include <stack>
#include <set>
#include <numeric>
#include <sstream>
#include <iomanip>
#include <limits>

#define CLR(a) memset(a, 0, sizeof(a))

using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
typedef pair<string, int> psi;
typedef pair<string, string> pss;

const double PI = 3.14159265358979323846264338327;
const double E = exp(1);
const double eps = 1e-6;

const int INF = 0x3f3f3f3f;
const int maxn = 1e3 + 5;
const int MOD = 1e9 + 7;

int main()
{
    map <int, int> vis;
    int n, m, num;
    scanf("%d%d", &n, &m);
    vector <int> v;
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &num);
        num = -num;
        v.push_back(num);
        push_heap(v.begin(), v.end());
    }
    for (int i = 0; i < n; i++)
        vis[-v[i]] = i + 1;
    while (m--)
    {
        int a, b, c, d;
        string s;
        cin >> a;
        cin >> s;
        if (s == "and")  //2
        {
            cin >> b;
            cin >> s;
            cin >> s;
            c = vis[a];
            d = vis[b];
            if (v[c / 2] == v[d / 2] && c != d)
                cout << "T\n";
            else
                cout << "F\n";
        }
        else
        {
            cin >> s;
            if (s == "the")
            {
                cin >> s;
                if (s == "root")  //1
                {
                    if (vis[a] == 1)
                        cout << "T\n";
                    else
                        cout << "F\n";
                }
                else    // 3
                {
                    cin >> s;
                    cin >> b;
                    c = vis[a];
                    d = vis[b];
                    if (d / 2 == c)
                        cout << "T\n";
                    else
                        cout << "F\n";
                }
            }
            else  //4
            {
                cin >> s;
                cin >> s;
                cin >> b;
                c = vis[a];
                d = vis[b];
                if (c / 2 == d)
                    cout << "T\n";
                else
                    cout << "F\n";
            }
        }
    }
}