首页 > 代码库 > 堆or优先队列(洛谷1801 黑匣子_NOI导刊2010提高（06）)
堆or优先队列(洛谷1801 黑匣子_NOI导刊2010提高（06）)
Black Box是一种原始的数据库。它可以储存一个整数数组，还有一个特别的变量i。最开始的时候Black Box是空的．而i等于0。这个Black Box要处理一串命令。
命令只有两种：
ADD(x):把x元素放进BlackBox;
GET:i加1，然后输出Blackhox中第i小的数。
记住：第i小的数，就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如：
我们来演示一下一个有11个命令的命令串。（如下图所示）
现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多200000个。现在用两个整数数组来表示命令串：
1.A(1)，A(2)，…A(M)：一串将要被放进Black Box的元素。每个数都是绝对值不超过2000000000的整数，M$200000。例如上面的例子就是A=(3，1，一4，2，8，-1000，2)。
2.u(1)，u(2)，…u(N)：表示第u(j)个元素被放进了Blaek Box里后就出现一个GET命令。例如上面的例子中u=(l，2，6，6)。输入数据不用判错。
输入格式：
第一行，两个整数，M，N。
第二行，M个整数，表示A(l)
……A(M)。
第三行，N个整数，表示u(l)
…u(N)。
输出格式：
输出Black Box根据命令串所得出的输出串，一个数字一行。
输入样例#1：
7 43 1 -4 2 8-1000 21 2 6 6
输出样例#1：
3312
思路：创建两个堆，分别为大根堆和小根堆，然后使得大根堆里的数的个数保持为k，并且大根堆的所有元素都是小于小根堆的，这样每次取大根堆的堆顶就是答案了~
#include<bits/stdc++.h>#define maxn 400000using namespace std;int n,m;int a[maxn],u[maxn];struct cmp{ bool operator () (const int a,const int b) const { return a<b; }};priority_queue<int ,vector<int>,greater<int> >q;priority_queue<int ,vector<int>,cmp >que;int main(){ ios::sync_with_stdio(false); cin>>m>>n; for(int i=1;i<=m;i++) cin>>a[i]; for(int i=1;i<=n;i++) cin>>u[i]; int k=1;//k标记的是在u数组中的指针的地址 ，也是get第几位 for(int i=1;i<=m;i++) { que.push(a[i]); if(que.size()>0) { q.push(que.top()); que.pop(); } while(u[k]==i)//通过这句使得que大根堆中的数始终保持k-1个 { cout<<q.top()<<endl; que.push(q.top()); q.pop(); k++; } } return 0;}
堆or优先队列(洛谷1801 黑匣子_NOI导刊2010提高（06）)
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。