POJ 1442 Black Box(第k次询问第k小,两个堆)

思路:

用一个大顶堆和一个小顶堆,将队列分为两部分,大顶堆的堆顶小于小顶堆的堆顶,这样就连续的把数列分开了,每次询问只要把小顶堆里的数扔到大顶堆里去,然后调整两个堆使得大顶堆的堆顶小于小顶堆的堆顶
,然后此时大顶堆的堆顶就是第k小。


#include <iostream>
#include <cstdio>
#include <string.h>
#include <queue>
#include <cmath>
typedef long long int lli;
using namespace std;

int a[30100];
int q[30100];

int main(){
    int m,n;
    cin>>m>>n;
    priority_queue<int , vector<int>,greater<int> > q1;
    priority_queue<int> q2;

    for(int i = 1;i <= m;i++){
        scanf("%d",a+i);
    }
    for(int i = 1;i <= n;i++){
        scanf("%d",q+i);
    }

    int i = 0;
    for(int j = 1;j <= n;j++){
        while(i < q[j]){
            i++;
            q1.push(a[i]);
        }
        int temp = q1.top();
        q1.pop();
        q2.push(temp);
        while(!q1.empty() && q2.top() > q1.top()){
            q1.push(q2.top());
            q2.pop();
            q2.push(q1.top());
            q1.pop();
        }
        printf("%d\n",q2.top());
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值