STL 之 heap

heap是一个满二叉树，实际运用中可以化为一个数组。

把数组ans[]化为heap之后，其第i个元素的父节点（假如存在）是i/2，其左儿子为2*i，右儿子为2*i+1。

heap默认是最大堆。

下图是从STL-源码剖析中copy出来的：

heap是一个满二叉树，上图是最大堆

 

 

heap有关的函数列表

make_heap	把数组或vector化为heap make_heap(ans, ans+sizeof(ans)/sizeof(ans[0]), cmp=less<>())[数组]或 make_heap(ans.begin(), ans.end(), cmp=less<>())[vector] 默认是最大堆，也可以自己定义cmp改变规则
push_heap	只能对vector进行操作，因为在进行此操作前，首先得在尾端添加元素 vec.push_back(val); push_heap(vec.begin(), vec.end(), cmp=less<>());
pop_heap	把堆顶元素置为堆尾 vec.pop_heap(vec.begin(), vec.end(), cmp=less<>()); 默认最大堆
sort_heap	1,如果堆是最大堆，则应非减排序；如果堆是最小堆，则应非增排序；否则就会排序出错，出现堆是无效堆 2,sort_heap(vec.begin(),vec.end(), cmp=greater<>())

 

以下是实践代码：

 

#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <iterator>
#include <ctime>
#include <functional>
 
using namespace std;
const int MAX_N = 5;
 
void O(vector<int>vec, char* s)
{
cout << s << endl;
copy(vec.begin(), vec.end(), ostream_iterator<int>(cout, " "));
cout << endl << endl;
}
 
int main()
{
vector<int>vec;
int i, j, k;
 
srand(time(NULL));
for (i = 0; i < MAX_N; i++)
vec.push_back(rand()%1000);
O(vec, "----init vector----");
 
make_heap(vec.begin(), vec.end(), greater<int>());
O(vec, "----after make_heap----");
 
sort_heap(vec.begin(), vec.end(), greater<int>());
O(vec, "----after sort_heap----");
 
i = rand()%1000;
cout << "the value of push_back = " << i << endl;
vec.push_back(i);
push_heap(vec.begin(), vec.end(), greater<int>());
O(vec, "----after push_heap----");
 
pop_heap(vec.begin(), vec.end());
O(vec, "----after pop_head----");
 
pop_heap(vec.begin(), vec.end());
O(vec, "----after pop_head----");
 
exit(0);
}