C++如何使用priority_queue_C++优先队列容器与应用示例

priority_queue是C++ STL中基于堆的容器适配器，默认为最大堆，常用于任务调度、Dijkstra算法等场景。1. 基本用法：包含头文件，定义方式有priority_queue pq（最大堆）和priority_queue pq（最小堆），常用操作包括push、pop、top、empty和size。示例输出为30 20 10。2. 自定义比较：可通过重载operator

在C++中，priority_queue 是标准模板库（STL）提供的一个容器适配器，用于实现优先队列。它基于堆结构，默认情况下使用最大堆，即每次取出的元素是当前队列中最大的。它常用于需要按优先级处理数据的场景，比如任务调度、Dijkstra最短路径算法等。

1. priority_queue的基本用法

头文件： 使用 #include 即可引入 priority_queue。

定义方式：

• priority_queue pq;：默认最大堆，顶部是最大值。
• priority_queue, greater> pq;：最小堆，顶部是最小值。

常用操作：

立即学习“C++免费学习笔记（深入）”；

• pq.push(x)：插入元素 x。
• pq.pop()：移除顶部元素（不返回）。
• pq.top()：获取顶部元素。
• pq.empty()：判断是否为空。
• pq.size()：返回元素个数。

示例代码：

#include 
#include 
using namespace std;
int main() {
priority_queue pq;
pq.push(10);
pq.push(30);
pq.push(20);
while (!pq.empty()) {
    cout << pq.top() << " ";
    pq.pop();
}
// 输出：30 20 10
return 0;
}

2. 自定义比较函数（结构体或类）
当需要处理自定义类型（如结构体）时，可以通过重载比较运算符或提供比较类来定义优先级规则。
方法一：重载 operator

							

								
								

									谱乐AI
									
谱乐AI，集成 Suno、Udio 等顶尖AI音乐模型的一站式AI音乐生成平台。
								
								下载 
							
						
struct Person {
    string name;
    int age;
    Person(string n, int a) : name(n), age(a) {}
bool operator<(const Person& p) const {
    return age < p.age; // 年龄大的优先级高（最大堆）
}
};
priority_queue pq;
pq.push(Person("Alice", 25));
pq.push(Person("Bob", 30));
cout 
方法二：自定义比较结构体
struct CompareAge {
    bool operator()(const Person& a, const Person& b) {
        return a.age < b.age; // 最大堆
    }
};
priority_queue, CompareAge> pq;
若想实现年龄小的优先，则改为：a.age > b.age，即可变为最小堆逻辑。
3. 实际应用示例：合并K个有序链表
LeetCode经典题：使用 priority_queue 维护每条链表当前最小节点，每次取出最小值节点加入结果链表。
#include 
struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(nullptr) {}
};
struct Compare {
bool operator()(ListNode a, ListNode b) {
return a->val > b->val; // 最小堆：值小的优先
}
};
ListNode mergeKLists(vector>& lists) {
priority_queue, vector>, Compare> pq;
for (ListNode* head : lists) {
    if (head) pq.push(head);
}

ListNode dummy(0);
ListNode* tail = &dummy;

while (!pq.empty()) {
    ListNode* node = pq.top(); pq.pop();
    tail->next = node;
    tail = tail->next;
    if (node->next) pq.push(node->next);
}

return dummy.next;
}

该方法时间复杂度为 O(N log k)，其中 N 是所有节点总数，k 是链表数量，效率较高。
4. 注意事项与常见问题


priority_queue 不支持遍历，只能访问 top 元素。
修改已有元素会破坏堆结构，应避免。
如果需要动态更新优先级，考虑使用 set 或手写堆。
注意自定义比较函数的逻辑方向，greater 表示“更小的优先”，即最小堆。

基本上就这些。掌握 priority_queue 能显著简化很多算法设计，特别是在贪心和图算法中非常实用。熟悉其接口和自定义方式后，使用起来很高效。