PAT A1147 Heaps (30 分) 堆

最新推荐文章于 2020-09-02 18:27:37 发布

HackingGate

最新推荐文章于 2020-09-02 18:27:37 发布

阅读量124

点赞数

分类专栏： PAT

本文链接：https://blog.csdn.net/chch1996/article/details/100633165

版权

PAT 专栏收录该内容

165 篇文章 0 订阅

订阅专栏

题目大意：给出完全二叉树的层序遍历序列，判断这棵完全二叉树是否是最大或最小堆，并输出后序遍历的结果。

以下标为0为根节点，下标为k的节点的左孩子下标是 2*k+1, 右孩子下标是 2*k+2，最后一个非叶子节点的下标是（N-2)/2。因此遍历下标为 0 ~ (N-2) / 2 的节点，考察它们与左右孩子的大小关系，容易判断出是否是最大或最小堆。

AC代码：

#include <vector>
#include <algorithm>
#include <cstdio>

using namespace std;

bool judgeMaxHeap(vector<int> &v)
{
    for (int i = 0; i <= (v.size() - 1 - 1) / 2; ++i)
    {
        if(v[i] < v[2*i+1] || v[i] < v[min(2*i+2, (int)v.size() - 1)]) return false;
    }
    return true;
}

bool judgeMinHeap(vector<int> &v)
{
    for (int i = 0; i <= (v.size() - 1- 1) / 2; ++i)
    {
        if(v[i] > v[2*i+1] || v[i] > v[min(2*i+2, (int)v.size() - 1)]) return false;
    }
    return true;
}

void postOrder(int node, vector<int> &v, int &printCnt)
{
    if(node > v.size() - 1) return;
    postOrder(node*2+1, v, printCnt);
    postOrder(node*2+2, v, printCnt);
    printf("%d", v[node]);
    if(++printCnt < v.size()) printf(" ");
    else printf("\n");
}

int main()
{
    int M, N;
    scanf("%d%d", &M, &N);
    for (int query = 0; query < M; ++query)
    {
        vector<int> v(N);
        for (int i = 0; i < N; ++i)
        {
            scanf("%d", &v[i]);
        }
        bool isMaxHeap = judgeMaxHeap(v);
        if(isMaxHeap) printf("Max Heap\n");
        else
        {
            bool isMinHeap = judgeMinHeap(v);
            if(isMinHeap) printf("Min Heap\n");
            else printf("Not Heap\n");
        }
        int printCnt = 0;
        postOrder(0, v, printCnt);
    }
    return 0;
}