C++中的priority_queue
是一个容器适配器,它提供了常数时间查找最大元素(在默认情况下)和对数时间删除最大元素的能力。这使得它非常适合于实现贪心算法、Dijkstra算法和A*算法等需要优先级队列的场景。
以下是priority_queue
在算法中的一些应用:
priority_queue
来存储待处理的元素,并在每一步选择优先级最高(通常是最小或最大)的元素进行处理。priority_queue
来存储待处理的顶点,并在每一步选择距离源顶点最近的顶点进行处理。priority_queue
来存储待处理的节点,并在每一步选择具有最低估计成本的节点进行处理。priority_queue
默认实现的是最大堆,但你可以通过自定义比较函数来实现最小堆。最大堆和最小堆可以用于解决各种问题,如求解数组中的第k大/小元素、中位数查找等。下面是一个使用priority_queue
实现的简单示例,该示例找出数组中的前k个最大元素:
#include<iostream>
#include<queue>
#include<vector>
int main() {
std::vector<int> nums = {3, 2, 1, 5, 6, 4};
int k = 2;
std::priority_queue<int, std::vector<int>, std::greater<int>> pq;
for (int num : nums) {
pq.push(num);
if (pq.size() > k) {
pq.pop();
}
}
while (!pq.empty()) {
std::cout << pq.top() << " ";
pq.pop();
}
return 0;
}
在这个示例中,我们使用了一个最小堆(通过指定std::greater<int>
作为比较函数)来存储数组中的元素。我们遍历数组,将每个元素添加到优先级队列中,并在队列大小超过k时删除最小元素。最后,我们输出优先级队列中的所有元素,这些元素就是数组中的前k个最大元素。