C++ priority_queue
priority_queue介绍(优先级队列)
优先级队列是一种容器适配器,专门设计使其第一个元素始终是它包含的元素中最大的,根据一些严格的弱排序标准。
此上下文类似于堆,其中元素可以随时插入,并且只能检索最大堆元素(优先级队列中位于顶部的元素)。
优先级队列作为容器适配器实现,容器适配器是使用特定容器类的封装对象作为其基础容器的类,提供一组特定的成员函数来访问其元素。元素从特定容器的 “back” 弹出,这称为优先级队列的顶部。
priority_queue使用
优先级队列默认使用vector作为其底层存储数据的容器,在vector上又使用了堆算法将vector中元素构造成堆的结构,因此priority_queue就是堆,所有需要用到堆的位置,都可以考虑使用 priority_queue。注意:默认情况下priority_queue是大堆。
priority_queue()/priority_queue(first, last) 构造一个空的优先级队列
empty( ) 检测优先级队列是否为空,是返回true,否则返回false
top( ) 返回优先级队列中最大(最小元素),即堆顶元素
push(x) 在优先级队列中插入元素x
pop() 删除优先级队列中最大(最小)元素,即堆顶元素
vector<int> v{3,2,7,6,0,4,1,9,8,5};
priority_queue<int> q1;
for (auto& e : v)
q1.push(e);
cout << q1.top() << endl;
只要将数据存入优先级队列,优先级队列就能自动排序,优先级队列在默认情况下是大堆。
priority_queue<int, vector<int>, greater<int>> q2(v.begin(), v.end());
cout << q2.top() << endl;
如果要创建小堆,将第三个模板参数换成greater比较方式。
如果在priority_queue中放自定义类型的数据,用户需要在自定义类型中提供> 或者< 的重载。
练习
数组中第k个大的数
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int> q1;
for(int i=0;i<nums.size();i++)
{
q1.push(nums[i]);
}
for(int i=1;i<k;i++)
{
q1.pop();
}
return q1.top();
}
直接通过调用优先级队列在存储数组中的元素,随后pop掉数组中前k个大的数,再返回优先级队列的栈顶即可完成