C++如何实现一个堆(Heap)_C++中使用priority_queue实现最大堆和最小堆

C++中priority_queue默认实现最大堆,通过指定greater比较器可实现最小堆,支持自定义类型及比较逻辑,简化堆操作。

在C++中,堆(Heap)是一种特殊的完全二叉树结构,分为最大堆和最小堆。最大堆的根节点值最大,最小堆的根节点值最小。虽然可以手动实现堆的插入、删除等操作,但C++标准库提供了更简便的方式——priority_queue,它默认实现的是最大堆。

使用 priority_queue 实现最大堆

priority_queue 默认基于 vectorless 比较器,因此顶部元素是最大的,即最大堆。

  • #include
  • #include iostream>
  • std::priority_queue max_heap;
  • max_heap.push(10);
  • max_heap.push(30);
  • max_heap.push(20);
  • std::cout

每次调用 top() 获取当前最大值,pop() 删除最大值,所有操作自动维护堆结构。

使用 priority_queue 实现最小堆

要实现最小堆,需指定第三个模板参数为 greater,并明确容器类型。

  • #include
  • #include
  • #include // std::greater
  • std::priority_queue, std::greater> min_heap;
  • min_heap.push(30);
  • min_heap.push(10);
  • min_heap.push(20);
  • std::cout

此时堆顶始终是最小元素,适用于需要频繁获取最小值的场景,比如Dijkstra算法。

自定义类型如何使用 priority_queue

若堆中存储的是自定义结构体或类,需提供比较逻辑。可通过重载操作符或传入仿函数。

  • struct Person {
  •   int age;
  •   std::string name;
  • };
  • // 最小堆:按年龄排序
  • auto cmp = [](const Person& a, const Person& b) { return a.age > b.age; };
  • std::priority_queue, decltype(cmp)> pq(cmp);

使用lambda表达式作为比较器时,需将其实例传入构造函数,并在模板中声明类型。

基本上就这些。priority_queue 封装了堆的核心操作,避免手动调整堆结构,提升开发效率。理解其默认行为和如何切换最小堆,能灵活应对各类算法需求。不复杂但容易忽略细节,比如最小堆必须显式指定容器和比较器。