您的位置:首页 > 编程语言 > C语言/C++

POJ 1442 Black Box(优先队列)

2014-08-04 21:11 344 查看
题目地址:POJ 1442

这题是用了两个优先队列,其中一个是较大优先,另一个是较小优先。让较大优先的队列保持k个。每次输出较大优先队列的队头。

每次取出一个数之后,都要先进行判断,如果这个数比较大优先的队列的队头要小,就让它加入这个队列,队列头移到较小优先的队列中。然后当较大优先的数不足k个的时候,就让较小优先的队列的队头移到较大优先的队头中。

代码如下;

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
int a[40000], b[40000];
struct xiao
{
int x;
bool operator < (const xiao &a) const
{
return x > a.x;
}
};
struct da
{
int x;
bool operator < (const da &a) const
{
return x<a.x;
}
};
int main()
{
int n, m, i, j, cnt, x, k;
while(scanf("%d%d",&n,&m)!=EOF)
{
priority_queue<xiao>q1;
priority_queue<da>q2;
xiao f1;
da f2;
for(i=0; i<n; i++)
{
scanf("%d",&a[i]);
}
for(i=0; i<m; i++)
{
scanf("%d",&b[i]);
}
j=0;
for(i=0; i<m; i++)
{
while(j<b[i])
{
if(q2.empty()||a[j]>=q2.top().x)
{
f1.x=a[j];
q1.push(f1);
}
else
{
f1.x=q2.top().x;
q1.push(f1);
q2.pop();
f2.x=a[j];
q2.push(f2);
}
j++;
}
while(q2.size()<=i)
{
f1=q1.top();
f2.x=f1.x;
q1.pop();
q2.push(f2);
}
printf("%d\n",q2.top());
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息