C++ STL堆操作

/* STL 最大堆、最小堆的应用 */
#include <iostream>
#include <vector>
#include <algorithm>    //

using namespace std;

/*
STL 堆操作
(1)make_heap()构造堆
void make_heap(first_pointer,end_pointer,compare_function);
默认比较函数是(<),即最大堆。
函数的作用是将[begin,end)内的元素处理成堆的结构

(2)push_heap()添加元素到堆
void push_heap(first_pointer,end_pointer,compare_function);
新添加一个元素在末尾,然后重新调整堆序。该算法必须是在一个已经满足堆序的条件下。
先在vector的末尾添加元素,再调用push_heap

(3)pop_heap()从堆中移出元素
void pop_heap(first_pointer,end_pointer,compare_function);
把堆顶元素取出来,放到了数组或者是vector的末尾。
要取走,则可以使用底部容器(vector)提供的pop_back()函数。
先调用pop_heap再从vector中pop_back元素

(4)sort_heap()对整个堆排序
排序之后的元素就不再是一个合法的堆了。
*/

//最大堆
struct MaxHeapCmp
{
    inline bool operator()(const int &x,const int &y)
    {
        return x < y;
    }
};

//最小堆
struct MinHeapCmp
{
    inline bool operator()(const int &x, const int &y)
    {
        return x > y;
    }
};

void test()
{
    std::vector<int> data{ 3,1,7,4,99,75,34,10 };

    //堆排序
    std::make_heap(data.begin(), data.end(), MinHeapCmp());
    for (int n : data)
    {
        cout << n << endl;
    }

    printf("------------------
");

    //添加元素
    data.push_back(111);
    //再次堆排序
    std::push_heap(data.begin(), data.end(), MaxHeapCmp());
    for (int n : data)
    {
        cout << n << endl;
    }
}

int main()
{
    test();
    printf("-ok-
");
    getchar();

    return 0;
}
原文地址:https://www.cnblogs.com/zhanggaofeng/p/10657165.html