pku 1442 Black Box(优先队列)

http://acm.pku.edu.cn/JudgeOnline/problem?id=1442

 

题目大意：给你两个序列a,b，一个长度为N，一个长度为M。b中每个元素值c表示a序列中前c个数中，已经输入进来了，求输入进来的C个数中第k小的数。这个k就是c在序列b中的下标。

 

解题思路：优先队列做它。定义一个极大优先队列，用来保存前k个小的数，定义一个极小优先队列，用来保存后面剩下的c-k个数。最后输出极大队列的top值就好。

 

#include <queue> #include <iostream> #include <stdio.h> #define Max 30005 using namespace std; struct node { int val; node(int x){val = 0;} node(){;} }; int val[Max],p[Max]; bool operator<(node a,node b) { return a.val>b.val; } void read(int M,int N) { int i; for(i=1;i<=M;i++) scanf("%d",&val[i]); for(i=1;i<=N;i++) scanf("%d",&p[i]); } int main() { priority_queue<node>qmin;/*极小优先队列*/ priority_queue<int>qmax;/*极大优先队列*/ int M,N,i,j,tem; node a; while (scanf("%d%d",&M,&N)!=EOF) { read(M,N); for (i=1,j=1;i<=N;i++) { /*如果当前要求第i大的数，但是qmax中没有i个数，则从qmin中取出一个元素出来*/ if(!qmin.empty()&&qmax.size()<=i-1) { a = qmin.top(); qmin.pop(); qmax.push(a.val); } for (;j<=p[i];j++) { /*容量小于i，则直接入队*/ if (qmax.size()<=i-1) qmax.push(val[j]); else { /*当前值大于第i大数，肯定不会影响前i个数的顺序，插入到qmin队列中保存*/ if(val[j]>=qmax.top()) { a.val = val[j]; qmin.push(a); } else { /*如果有影响，则qmax.top排名肯定后移一位，为第i+1大的数*/ tem = qmax.top(); a.val = tem; qmax.pop(); qmax.push(val[j]); qmin.push(a); } } } printf("%d/n",qmax.top()); } /*清空队列*/ while(!qmin.empty()) qmin.pop(); while(!qmax.empty()) qmax.pop(); } return 0; }