您的位置 首页 编程知识

c++如何使用优先队列(priority_queue)_C++堆结构与优先队列用法详解

优先队列基于堆实现,默认为大根堆,包含于queue头文件中。使用std::priority_queue<…


优先队列基于堆实现,默认为大根堆,包含于queue头文件中。使用std::priority_queue<int>声明,默认提供push、top、pop等操作。通过greater<int>可创建小根堆:priority_queue<int, vector<int>, greater<int>>。自定义类型需重载<或定义比较结构体。常用于Top K、Dijkstra、合并K链表等场景。

c++如何使用优先队列(priority_queue)_C++堆结构与优先队列用法详解

在C++中,优先队列(priority_queue)是一种基于堆结构实现的容器适配器,能够自动将元素按优先级排序,默认情况下是大根堆,即最大元素始终位于队首。它定义在 queue 头文件中,使用非常方便,适用于需要动态维护最大值或最小值的场景,比如Dijkstra算法、合并K个有序链表等。

1. 基本用法与定义

要使用 priority_queue,需包含头文件:

#include <queue>

基本声明方式如下:

std::priority_queue<int> pq; // 默认大顶堆,最大的元素在顶部

插入和访问元素的操作:

立即学习“”;

  • pq.push(x):插入元素 x
  • pq.top():获取堆顶元素(最高优先级)
  • pq.pop():删除堆顶元素
  • pq.empty():判断是否为空
  • pq.size():返回元素个数

示例代码:

#include <tream>
#include <queue>
using namespace std;

int mn() {
priority_queue<int> pq;
pq.push(10);
pq.push(30);
pq.push(20);

while (!pq.empty()) {
cout << pq.top() << ” “; // 输出: 30 20 10
pq.pop();
}
return 0;
}

2. 使用小根堆(最小堆)

默认是大根堆,若想让最小元素在顶部,可以指定比较方式。C++ 提供了 greater 比较器:

std::priority_queue<int, vector<int>, greater<int>> min_pq;

说明:

数据为基,先见未见

c++如何使用优先队列(priority_queue)_C++堆结构与优先队列用法详解 95

  • 第一个参数:元素类型
  • 第二个参数:底层容器,通常为 vector<int>
  • 第三个参数:比较函数对象,greater<int> 表示小顶堆

示例:

priority_queue<int, vector<int>, greater<int>> min_pq;
min_pq.push(30);
min_pq.push(10);
min_pq.push(20);

while (!min_pq.empty()) {
cout << min_pq.top() << ” “; // 输出: 10 20 30
min_pq.pop();
}

3. 自定义数据类型与比较规则

如果要存储自定义结构体,比如任务优先级,就需要重载比较规则。有两种常用方法:

方法一:重载操作符 <

struct Task {
int priority;
string name;
bool operator<(const Task& t) const {
return priority < t.priority; // 优先级高的排前面(大根堆)
}
};

priority_queue<Task> task_pq;

方法二:自定义比较结构体

struct Compare {
bool operator()(const Task& a, const Task& b) {
return a.priority < b.priority; // 大根堆
}
};

priority_queue<Task, vector<Task>, Compare> task_pq;

注意:在 priority_queue 中,比较函数的作用是判断哪个元素“更小”,因此如果返回 true,表示 a 应该比 b 后弹出(即 a 优先级低)。

4. 常见应用场景

优先队列常用于以下问题:

  • 求前 K 大/小的元素(Top K 问题)
  • 合并 K 个有序链表
  • Dijkstra 最短路径算法
  • 哈夫曼
  • 任务调度系统

例如,求一个数组中最大的 K 个数,可以用小根堆维护大小为 K 的堆,遍历数组不断更新堆顶。

基本上就这些。掌握构造方式、自定义比较和基本操作,就能灵活使用 priority_queue 解决多数问题。

以上就是++如何使用优先队列(priority_queue)_C++堆结构与优先队列用法详解的详细内容,更多请关注php中文网其它相关文章!

相关标签:

大家都在看:

本文来自网络,不代表四平甲倪网络网站制作专家立场,转载请注明出处:http://www.elephantgpt.cn/16245.html

作者: nijia

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注

联系我们

联系我们

18844404989

在线咨询: QQ交谈

邮箱: 641522856@qq.com

工作时间:周一至周五,9:00-17:30,节假日休息

关注微信
微信扫一扫关注我们

微信扫一扫关注我们

关注微博
返回顶部