题目传送门: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--;
}
}