洛谷p1801黑匣子(双堆)

题意:对于一个数字序列,每次询问回答第a[i]个元素插入后的元素中的第j大值。
思路:由于询问是有序的,先将询问离线下来,开一个大根堆和一个小根堆,其中大根堆是严格保持k-1个元素的容器,每次元素数量==k时就把多出来的元素转移到小根堆中保存,这样保证每次回答都是小根堆的堆顶。(因为会进入小根堆是因为大根堆已经刚好满了,进入小根堆后又会在堆中重新堆排序,堆顶就是小根堆中最小值,即第k大值)

代码如下:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
priority_queue< ll, vector<ll>, greater<ll> >q1;//小根堆;
priority_queue<ll>q2;//大根堆; 
ll a[200005];//原数组; 
ll b[200005];//查询数组; 

int main()
{
	int i;
	int j;
	int n,m;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	}
	for(i=1;i<=m;i++)
	{
		scanf("%d",&b[i]);
	}
	while(!q1.empty())
	{
		q1.pop();
	}
	while(!q2.empty())
	{
		q2.pop();
	}
	j=1;
	for(i=1;i<=m;i++)//查询的序号; 
	{
		for(;j<=b[i];j++)//每次查询中所到元素个数; 
		{
			q2.push(a[j]);//把元素插入到大根堆中;
			if(q2.size()==i)//保持大根堆中元素数恒为i-1个; 
			{
				q1.push(q2.top());
				q2.pop();
			 } 
		}
		printf("%lld\n",q1.top());//小根堆中的堆顶元素即为整个目前系列中的第k大;(因为小根堆中的最小是大根堆中最大顶中的最小值);
		q2.push(q1.top());
		q1.pop(); 
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值