P1801 黑匣子（对顶堆）

最新推荐文章于 2025-06-07 01:31:25 发布

weixin_33937913

最新推荐文章于 2025-06-07 01:31:25 发布

阅读量190

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/dfsac/p/6819747.html

本文介绍了一种利用大根堆和小根堆的数据结构来解决寻找序列中第s小元素的问题。通过维护两个堆的状态，可以有效地输出序列中的指定元素，并通过实例代码展示了具体的实现过程。

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

https://www.luogu.org/problem/show?pid=1801#sub

用对顶堆来做：
上次输出第s小的数，在大根堆里储存s数（这个状态需要维护），在小根堆里存剩下的数，下一次输出时输出小根堆的顶，即是需要输出的值，s++

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
priority_queue <int> qued; 
priority_queue <int> quex;

int u[200001];
int n,m,a[200001];
int main()
{
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++)
      scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        u[x]++;
    }
    int s=0;
    for(int i=1;i<=m;i++)
    {
        qued.push(a[i]);
        if(qued.size()>s){
            quex.push(-qued.top());
            qued.pop();
        }
        while(u[i])
        {
            s++;
            printf("%d\n",-quex.top());
            qued.push(-quex.top());
            quex.pop();
            u[i]--;
        }

    }
    return 0;
}