C++ STL系列之:heap

实现

通过vector和完全二叉树实现。
建立堆make_heap(),在堆中添加数据push_heap(),在堆中删除数据pop_heap()和堆排序sort_heap():
头文件 #include
函数:

make_heap(start,end,cmp)
push_heap(start,end) //在vector中加好了(push_back),再使用这个函数重新调整一下堆
pop_heap() //使用后,root 结点就被放到了vector的最后,再使用vector的pop_back()删除
sort_heap(start,end,cmp)

这里需要注意的就是cmp的用法,默认情况下是 最小堆,当要使用从大到小排序时,就是要使用最大堆,这里再make_heap的时候,就要加入自定义的比较函数,否则前面定义的最小堆,后面又自定义的比较函数从大到小排,这样没有用。

另外这里要是vector内的元素自定义的,要使用自定义的比较函数,就要为这个vector重载 大于 小于的运算符,或者像下面这里定义一个全局的比较函数。这里可以参考https://blog.csdn.net/aguisy/article/details/5787257

#include
#include
#include

using namespace std;

bool cmp(const int& a, const int& b)
{
    return a > b;
}


int main()
{
    int a[10] = {1,5,2,6,3,7,8,4,9,0};
    vector<int> v1(a,a+10);
    //make_heap(v1.begin(),v1.end(),greater());
    //sort_heap(v1.begin(),v1.end(),greater());

    //make_heap(v1.begin(),v1.end(),less());
    //sort_heap(v1.begin(),v1.end(),less());

    make_heap(v1.begin(),v1.end(),cmp);
    sort_heap(v1.begin(),v1.end(),cmp);

    for(int i=0; i<10; i++){
        cout << v1[i]<<";";
    }

    cout<return 0;
}

你可能感兴趣的:(C++)