POJ 1442 Black Box 优先队列求前a[k]个数中第k大的数

最新推荐文章于 2022-10-27 15:24:14 发布

原创最新推荐文章于 2022-10-27 15:24:14 发布 · 448 阅读

0 ·

CC 4.0 BY-SA版权

POJ 同时被 3 个专栏收录

51 篇文章

订阅专栏

STL

4 篇文章

订阅专栏

优先队列

3 篇文章

订阅专栏

本文介绍了一种使用两个优先级队列（一个最大堆和一个最小堆）来维护数据流中连续数据段中位数的方法。通过不断调整两个队列间的平衡，确保能够快速获取到中位数值。

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

维护两个队列，

开A

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>

using namespace std;

int a[31000];

int main()
{
    int m, n;
    scanf("%d %d", &m, &n);
    int i, j;
    for(i = 0;i < m;i++){
        scanf("%d", &a[i]);
    }
    int k;
    priority_queue<int, vector<int>, less<int> >big;
    priority_queue<int, vector<int>, greater<int> >small;
    j = 0;
    int t;
    for(i = 1;i <= n;i++){
        scanf("%d", &k);
        while(j < k){
            small.push(a[j]);
            if(!big.empty()&&small.top() < big.top()){
                t = big.top();
                big.pop();
                big.push(small.top());
                small.pop();
                small.push(t);
            }
            j++;
        }
        printf("%d\n", small.top());
        big.push(small.top());
        small.pop();
    }

    return 0;
}