黑匣子_NOI导刊2010提高(06)(对顶堆)

QAQ
要查询第k大的值,维护一个大根堆和一个小根堆,他们的队首元素一样,也就是说小根堆里的元素都比大根堆里的大。维护大根堆里有k个元素那么下次找答案时只需要输出小根堆的优先元素即可,然后再将小根堆的优先元素踢入大根堆23333

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
priority_queue <long long>  quex;
priority_queue <long long>  qued;
long long s[999999];
int f[999999];
int main()
{
    int n,m;
    int k=0;

    scanf("%d%d",&n,&m);

    for(int i=1;i<=n;i++)
     {
        scanf("%lld",&s[i]);
     }

    for(int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        f[x]++;
    } 

    for(int i=1;i<=n;i++)
    {
        long long x=s[i];
        if(k==0) quex.push(-x);
        else 
        {
            long long tmp=qued.top();
            if(x<tmp) //如果当前元素比大根堆的优先元素小的话,将其踢入大根堆,在将大根堆的优先元素踢入小根堆
            {
                qued.push(x);
                qued.pop();
                quex.push(-tmp);
            }
            else
            {
                quex.push(-x); 
            }
        }
        while(f[i])
        {
            k++;
            f[i]--;
            long long tmp=-quex.top();//小根堆的优先元素即为答案
            quex.pop();
            qued.push(tmp);//把小根堆的优先元素踢入大根堆
            printf("%lld\n",tmp);
        }
    }

    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值