优先队列:priority_queue
priority_queue
是一个优先队列,优先级高的最先出队,默认最大值优先。内部实现为堆,因此出队和入队的时间复杂度均为
O
(
l
o
g
n
)
O(logn)
O(logn)。可以自定义优先级控制出队顺序,如果是数值,则也可以采用加负号的方式实现最小值优先,优先队列不支持删除堆中的指定元素,只可以删除堆顶元素,如果需要删除指定元素,则可以采用懒操作。使用priority_queue
时,需要引入头文件#include <queue>
-
push(x)
:x
入队 -
pop()
:出队 -
top()
:取队头 -
size()
:返回队中的元素个数 -
empty()
:判断对空,若为空则返回true
题目描述
和黑子代表一个原始数据库,保存一个整数数组和一个特殊的 i i i变量。在最初的时刻,黑盒子是空的, i = 0 i = 0 i=0。黑盒子处理一系列命令(事务),有以下两种类型的事务。
- ADD(x):将元素
x
放入黑盒子。 - GET:将 i i i增加1,并给出包含在黑盒子中的所有整数中第 i i i小的值。第 i i i小的值是黑盒子中按非降序排序后的第 i i i个位置的数字。
写一个有效的算法来处理给定的事务序列。ADD和GET事务的最大数量均为30000。用两个整数数组来描述事务的顺序。
输入:输入包含M
,N
输出:根据给定的事务顺序输出答案序列,每行一个数字。
#include <iostream>
#include <queue>
#include <memory>
using namespace std;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int>> q2;
int main()
{
int n,m,x;
scanf("%d %d",&m,&n);
unique_ptr<int[]> a(new int[m+1]());//智能指针,括号表示将数组初始化为0
for (int i = 1; i <= m; ++i) {
scanf("%d",&a[i]);
}
int cnt = 1;
for(int i = 1; i<=n; i++){
scanf("%d",&x);
while(cnt <= x){
if(!q1.empty() && a[cnt]<q1.top()){
q2.push(q1.top());//将q1队头入q2队
q1.pop();//出队
q1.push(a[cnt]);//入队
}
else{
q2.push(a[cnt]);
}
cnt++;
}
printf("%d\n",q2.top());
q1.push(q2.top());
q2.pop();
}
return 0;
}
这里用到智能指针unique_ptr
,可以在板块结束自动释放内存,这里也是一点尝试。unique_ptr<int[]> a(new int[m+1]())
创建一个包含m+1
个元素的int
数组,括号表示将数组初始化为0。
输入:
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
输出:
3
3
1
2
尝试
想到优先队列结构应该也可以用来排序,所以进行一些测试。
#include <iostream>
#include <queue>
#include <memory>
using namespace std;
priority_queue<int> q1;
int main()
{
int m;
scanf("%d",&m);
unique_ptr<int[]> a(new int[m]());//智能指针,括号表示将数组初始化为0
for (int i = 0; i < m; ++i) {
scanf("%d",&a[i]);
}
for (int i = 0; i < m; ++i) {
q1.push(a[i]);
}
for (int i = 0; i < m; ++i) {
cout << q1.top() <<" ";
q1.pop();
}
return 0;
}
输入:
4
6 3 5 1
输出:
6 5 3 1
成功!