对顶堆

神奇的对顶堆
动态维护第K小
O(logn)单次
在这里插入图片描述
维护第k小
如图元素上大下小,保证大根堆中有k-1个元素
若大根堆元素个数=k,将大根堆堆顶弹入小根堆
每次取小根堆的堆顶即为答案。
例题:
洛谷P1801黑匣子
题目链接
代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[200005];
int m,n,t,s=1,k;
priority_queue<ll> q1;
priority_queue<ll,vector<ll>,greater<ll> >q2;
int main()
{
	cin>>m>>n;
	for(int i=1;i<=m;i++)cin>>a[i];
	while(n--)
	{
		cin>>t;
		k++;
		for(;s<=t;s++)
		{
			q1.push(a[s]);
			if(q1.size()==k)
			{
				q2.push(q1.top());
				q1.pop();
			}
		}
		cout<<q2.top()<<endl;
		q1.push(q2.top());
		q2.pop();
		s=t+1;
	}
	return 0;
 } 

洛谷P1168中位数
题目链接
代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[100005];
int m,n,t,s=1,k;
priority_queue<ll> q1;
priority_queue<ll,vector<ll>,greater<ll> >q2;
int main()
{
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>a[i];
		q1.push(a[i]);
		if(q1.size()>=i/2+1)
		{
			q2.push(q1.top());
			q1.pop();
		}
		if(i&1)
		{
			cout<<q2.top()<<endl;
			q1.push(q2.top());
			q2.pop();
		}
	}
	return 0;
 } 
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值