C++ STL系列之:heap

实现

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

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的时候,就要加入自定义的比较函数,否则前面定义的最小堆,后面又自定义的比较函数从大到小排,这样没有用。
感谢Irving_87指出错误,这里说的有问题。
默认情况下是最大堆,sort_heap()是每次拿出root放在vector末尾,然后再将前面的进行堆调整,形成新的堆,经sort_heap后,堆的结构就被破坏了。

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

#include<vector>
#include<iostream>
#include<algorithm>

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<int>());
    //sort_heap(v1.begin(),v1.end(),greater<int>());

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

    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<<endl;

    return 0;
}

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值