实现
通过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;
}