luogu P1801 黑匣子_NOI导刊2010提高(06)

题目传送门:https://www.luogu.org/problemnew/show/P1801#sub



题意:

有n个数依次读入,在第x个数被读入后会有一个输出操作,表示输出当前数列第i大的数,然后i++(此时的i是下一次输出操作的i)。



思路:

套路题,不细讲。

开两个堆,大根堆维护1~i-1小的数,小根堆维护剩下个数,每一次插入到小根堆里;对于输出操作,我们只需输出小根堆的堆顶,然后,因为i++,所以我们就要将小根堆的堆顶放入大根堆。注意,如果我们发现当前加入小根堆时的堆顶比大根堆的堆顶大,就交换它们(因为大根堆始终存的是1~i-1小的数)。



代码:

#include<cstdio>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > a;
priority_queue<int> b;
	int n,m;
	int p[300000],d[300000];
int main()
{
	int x;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&p[i]);
	for(int i=1;i<=m;i++)
		scanf("%d",&d[i]);
	int l1=1,l2=0;
	for(int i=1;i<=n;i++)
	{
		a.push(p[i]);
		l1++;
		if(l2)
		{
			int t1=a.top(),t2=b.top();
			if(t1<t2)
			{
				a.pop();
				b.pop();
				a.push(t2);
				b.push(t1);
			}
		}
		l2++;
		while(l2<=n&&i==d[l2])
		{
			b.push(a.top());
			a.pop();
			printf("%d\n",b.top());
			l1--;
			l2++;
		}
		l2--;
	}
}

阅读更多
个人分类:
上一篇莫队详解
下一篇C++卡常
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭