std::ranges::push_heap
来自cppreference.com
| 在标头 <algorithm> 定义
|
||
| 调用签名 |
||
| |
(1) | (C++20 起) |
| |
(2) | (C++20 起) |
将指定范围的末尾元素插入到某个关于 comp 和 proj 的堆中，该堆已包含该范围内除末尾元素外的所有元素。插入元素后的堆即是整个范围。
1) 指定的范围是
[first, last)。2) 指定的范围是
r。如果指定范围（除末尾元素外）不是关于 comp 和 proj 的堆，那么行为未定义。
此页面上描述的函数式实体是算法函数对象（非正式地称为 niebloid），即：
参数
| first, last | - | 要修改的元素范围的迭代器-哨位对 |
| r | - | 要修改的元素 range
|
| comp | - | 应用到投影后元素的比较器 |
| proj | - | 应用到元素的投影 |
返回值
1)
last2)
ranges::end(r)复杂度
最多应用 log(N) 次 comp 和 2log(N) 次 proj，其中 N 是：
1)
ranges::distance(first, last)2)
ranges::distance(r)可能的实现
struct push_heap_fn
{
template<std::random_access_iterator I, std::sentinel_for<I> S,
class Comp = ranges::less, class Proj = std::identity>
requires std::sortable<I, Comp, Proj>
constexpr I operator()(I first, S last, Comp comp = {}, Proj proj = {}) const
{
const auto n{ranges::distance(first, last)};
const auto length{n};
if (n > 1)
{
I last{first + n};
n = (n - 2) / 2;
I i{first + n};
if (std::invoke(comp, std::invoke(proj, *i), std::invoke(proj, *--last)))
{
std::iter_value_t<I> v {ranges::iter_move(last)};
do
{
*last = ranges::iter_move(i);
last = i;
if (n == 0)
break;
n = (n - 1) / 2;
i = first + n;
}
while (std::invoke(comp, std::invoke(proj, *i), std::invoke(proj, v)));
*last = std::move(v);
}
}
return first + length;
}
template<ranges::random_access_range R,
class Comp = ranges::less, class Proj = std::identity>
requires std::sortable<ranges::iterator_t<R>, Comp, Proj>
constexpr ranges::borrowed_iterator_t<R>
operator()(R&& r, Comp comp = {}, Proj proj = {}) const
{
return (*this)(ranges::begin(r), ranges::end(r), std::move(comp), std::move(proj));
}
};
inline constexpr push_heap_fn push_heap{};
|
示例
运行此代码
#include <algorithm>
#include <cmath>
#include <iostream>
#include <vector>
void out(const auto& what, int n = 1)
{
while (n-- > 0)
std::cout << what;
}
void print(auto rem, auto const& v)
{
out(rem);
for (auto e : v)
out(e), out(' ');
out('\n');
}
void draw_heap(auto const& v)
{
auto bails = [](int n, int w)
{
auto b = [](int w) { out("┌"), out("─", w), out("┴"), out("─", w), out("┐"); };
if (!(n /= 2))
return;
for (out(' ', w); n-- > 0;)
b(w), out(' ', w + w + 1);
out('\n');
};
auto data = [](int n, int w, auto& first, auto last)
{
for (out(' ', w); n-- > 0 && first != last; ++first)
out(*first), out(' ', w + w + 1);
out('\n');
};
auto tier = [&](int t, int m, auto& first, auto last)
{
const int n{1 << t};
const int w{(1 << (m - t - 1)) - 1};
bails(n, w), data(n, w, first, last);
};
const int m{static_cast<int>(std::ceil(std::log2(1 + v.size())))};
auto first{v.cbegin()};
for (int i{}; i != m; ++i)
tier(i, m, first, v.cend());
}
int main()
{
std::vector<int> v{1, 6, 1, 8, 0, 3,};
print("source vector v: ", v);
std::ranges::make_heap(v);
print("after make_heap: ", v);
draw_heap(v);
v.push_back(9);
print("before push_heap: ", v);
draw_heap(v);
std::ranges::push_heap(v);
print("after push_heap: ", v);
draw_heap(v);
}
输出：
source vector v: 1 6 1 8 0 3
after make_heap: 8 6 3 1 0 1
8
┌─┴─┐
6 3
┌┴┐ ┌┴┐
1 0 1
before push_heap: 8 6 3 1 0 1 9
8
┌─┴─┐
6 3
┌┴┐ ┌┴┐
1 0 1 9
after push_heap: 9 6 8 1 0 1 3
9
┌─┴─┐
6 8
┌┴┐ ┌┴┐
1 0 1 3
参阅
(C++20) |
检查给定范围是否为最大堆 (算法函数对象) |
(C++20) |
查找能成为最大堆的最大子范围 (算法函数对象) |
(C++20) |
从元素范围创建最大堆 (算法函数对象) |
(C++20) |
移除最大堆中最大元 (算法函数对象) |
(C++20) |
将最大堆变成按升序排序的元素范围 (算法函数对象) |
| 添加元素到最大堆 (函数模板) |