POJ 1442 Black Box

最新推荐文章于 2023-03-18 17:35:17 发布

拔剑吧炮姐是我的

最新推荐文章于 2023-03-18 17:35:17 发布

阅读量373

点赞数

CC 4.0 BY-SA版权

文章标签：优先队列 poj 简单题

本文链接：https://blog.youkuaiyun.com/qq_37328747/article/details/54426034

这篇博客介绍了如何利用优先队列解决POJ 1442题目的黑盒子问题。通过维护一个大顶堆和一个小顶堆，将数据按顺序加入，并在适当时候交换堆顶元素，从而实现找到每个位置上的第i小元素。代码中展示了具体的实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

**题目大意**
有一个黑盒子（Black Box）很神奇，可以获取其中第i小的元素。题目中有两种操作分别为ADD 和 GET，ADD操作序列为A，GET操作序列为u,GET初始值 i 为0，每次进行GET操作后 i 值加1。输出每次的第i小的值为多少。
**解题思路**
优先队列的简单应用，维护两个堆，一个大顶堆，一个小顶堆，将前u（p）个数据先加入小顶堆中，每次读取一个u（p）元素就将小顶堆的堆顶元素加入大顶堆中，当向小顶堆加入元素时要加入判断，判断小顶堆堆顶是否大于大顶堆堆顶，若否，则将两个堆顶元素互换。
**代码**
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <iostream>
using namespace std;
int num[50010];
int com[50010];
int main()
{
    int m,n,last,now,key1,key2;
    priority_queue<int,vector<int>,greater<int> > smallheap;
    priority_queue<int,vector<int>,less<int> > bigheap;
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++)
        scanf("%d",&num[i]);
    com[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%d",&com[i]);
        for(int j=1;j<=(com[i]-com[i-1]);j++){
            smallheap.push(num[com[i-1]+j]);
            if(!bigheap.empty()&&!smallheap.empty()&&smallheap.top()<bigheap.top()){
                key1=smallheap.top();key2=bigheap.top();
                smallheap.pop();bigheap.pop();
                smallheap.push(key2);bigheap.push(key1);
            }
        }
        printf("%d\n",smallheap.top());
        key1=smallheap.top();
        smallheap.pop();
        bigheap.push(key1);
    }
    return 0;
}