算法训练营 训练 黑盒子(优先队列)

2022/2/5 17:12:43

本文主要是介绍算法训练营 训练 黑盒子(优先队列),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

优先队列: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

成功!



这篇关于算法训练营 训练 黑盒子(优先队列)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程