建堆[HihoCoder-1405]

Building Heap HihoCoder-1405

hihoCoder太阁最新面经算法竞赛11

问题大意：给定一个$N$个元素的数组$A$（元素互不相同），要求你建立满足下列要求的二叉树$T$，并输出其前序遍历：

1）$T$满足最小堆性质；

2）输入的数组$A$满足$T$的中序遍历。

这是一道数据结构基础题，主要是解决如何建堆。题目要求最小堆，其树根必然是整个树的最小值。因此在建堆过程中，只需寻找最小值，根据找到的最小值的位置将中序遍历的序列一分为二，继续对这两个子序列建堆，直到子序列没有元素为止。利用递归很容易写出程序。这道题要求输出前序遍历结果，因此可以不用实际建堆而直接在建堆的同时输出即可。时间复杂度为$O(n m{log}n)$。

#include<iostream>
using namespace std;
template<typename T>
int findmin(T A[], int size)
{
    int min = 0;
    for (int i = 1; i < size; i++)
        if (A[i] < A[min])
            min = i;
    return min;
}
void bheap(int A[], int size)
{
    int minpos = findmin(A, size);
    cout << A[minpos] << " ";
    if (minpos!=0)
        bheap(A, minpos);
    if (minpos != size - 1)
        bheap(A + minpos + 1, size - minpos - 1);;
}
int main()
{
#define TMAX 150
    int A[TMAX];
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
        std::cin >> A[i];
    bheap(A, n);
    return 0;
}