【题解】洛谷P1801 黑匣子（堆）

最新推荐文章于 2024-07-15 09:52:00 发布

原创最新推荐文章于 2024-07-15 09:52:00 发布 · 272 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#题解 #洛谷 #堆

题解同时被 2 个专栏收录

140 篇文章

订阅专栏

堆

6 篇文章

订阅专栏

本文介绍了一种使用双堆（大根堆与小根堆）的数据结构方法来求解给定数组中第k小的数值。通过不断调整两堆的元素，确保小根堆的顶部始终是最新的第k小的数。

怎么求一个堆内的第k小的值呢？我们可以把问题化为两个堆来解决。

存一个大根堆一个小根堆，小根堆负责读入元素。a数组记录要读的元素，b数组记录当前位置。读入当前的范围内饱含的元素，并把这个元素放到小根堆里，如果大根堆非空（注意！！！）并且大根堆堆顶比小根堆堆顶要大就交换堆顶，最后输出小根堆堆顶，把它堆顶扔到大根堆里，代表已经找到了第i小的数。关于大根堆里的数是否能更新小根堆上面已经操作过了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int m,n;
int a[200010];
priority_queue<int,vector<int>,greater<int> > q;
priority_queue<int,vector<int>,less<int> > p;
int main()
{
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&a[i]);
	}
	int cnt=0;
	int w;
	for(int i=1;i<=n;i++)
	{
		
		scanf("%d",&w);
		while(cnt<w)
		{
			++cnt;
			q.push(a[cnt]);
			if(!p.empty()&&p.top()>a[cnt])
			{
				q.push(p.top());
				p.pop();
				p.push(a[cnt]);
				q.pop();
			}
		}
		
		printf("%d\n",q.top());
		p.push(q.top());
		q.pop();
		
	}
	return 0;
}