C++ 中的 heap 并不是一个单独的 STL 容器,而是基于 vector 的一系列算法。需要引入头文件
#include <algorithm>
构造堆
使用 make_heap(beg, end)
将迭代器范围 [beg, end)
内的元素调整为一个堆( beg
为堆顶)。
vector<int> v {5, 3, 6, 4, 7, 2, 1};
make_heap(v.begin(), v.end()); // v = {7, 5, 6, 4, 3, 2, 1}
make_heap
默认将数组序列调整为最大堆,可以使用第三个参数来指定推的类型。
vector<int> v{ 5, 3, 6, 4, 7, 2, 1 };
auto cmp = [](const int &n1, const int &n2) { return n1 > n2; };
make_heap(v.begin(), v.end(), cmp); // v = {1, 3, 2, 4, 7, 5, 6}
向堆添加元素
vector<int> v {5, 3, 6, 4, 7, 2, 1};
make_heap(v.begin(), v.end()); // v = {7, 5, 6, 4, 3, 2, 1}
v.push_back(8);// v = {7, 5, 6, 4, 3, 2, 1