洛谷 P1801 黑匣子（优先队列）

传送门

---

解题思路

这一道题和中位数那道题很像，如果不会中位数请先看一下那道题——中位数

这道题和那道题不同的地方在于：

那道题的第一个大根堆队列的数字数量是数字总数的一半，是根据数学知识计算出来的，并且输出是每两个数输出一个；

这道题的第一个大根堆队列的数字数量是get(i)中的i，每用一次get函数i就会加一，而且输出是碰到get在指定位置输出。

——基本没怎么两样。

注意维护时不要忘记和小根堆或大根堆的堆顶进行比较，确定是把当前元素放在小根堆中还是放在大根堆中。

AC代码

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int maxn=200005;
int m,n,g[maxn];
long long a[maxn];
priority_queue<long long> q1;
priority_queue<long long,vector<long long>,greater<long long> > q2;
int main()
{
    cin>>m>>n;
    for(int i=1;i<=m;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++) scanf("%d",&g[i]);
    int cnt=1;
    for(int i=1;i<=n;i++){
        for(;cnt<=g[i];cnt++){
            if(q1.empty()||a[cnt]<=q1.top()) q1.push(a[cnt]);
            else q2.push(a[cnt]);
            if(q1.size()>i){
                q2.push(q1.top());
                q1.pop();
            }
        }
        while(q1.size()<i){
            q1.push(q2.top());
            q2.pop();
        }
        printf("%lld\n",q1.top());
    }
    return 0;
}