官方参考文档:priority_queue - C++ Reference - cplusplus.com
无意间搜到这篇文章,十分喜欢作者对 Priority Queues 的介绍,特此翻译过来分享给大家。
priority queue 是一个不遵循先入先出规则的队列。
在 priority queue 中,我们可以添加连续的数据 ,并在常数时间内得到具有最高优先级的数据。
因此,要实现一个 priority queue,我们需要指定元素之间的比较规则来确定哪个元素具有较高的优先级。
在 C++ 中,我们可以使用 std::priority_queue
来包装另一个容器(默认是 std::vector
)。std::priority_queue
默认使用 std::less
来比较元素。因此具有最高优先级的元素是 最大的 那个元素。
std::priority_queue
也允许我们自定义比较方式。比如,我们可以使用 std::greater
,这样的话,具有最高优先级的元素是 最小的。我们也可以用其他的自定义比较符来比较自定义类型等。
std::priority_queue
提供了优先队列的接口:
- push:在队列中添加新的元素
- pop:移除队列中优先级最高的元素
- top:访问队列中优先级最高的元素
什么时候优先队列有用?一个例子是处理具有不同优先级的传入事件。我们希望根据事件的优先级处理事件,而不是根据事件的到达顺序。
堆实际上是优先队列的一种实现方式。
关于优先队列的运用,可以参考一下 Leetcode 数组中的第K个最大元素
该题可以使用自定义比较方式来用优先队列来模拟一个最小堆来完成。
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int, vector<int>, greater<int>> q;
for (int num : nums) {
q.push(num);
if (q.size() > k) {
q.pop();
}
}
return q.top();
}