题目链接:
https://www.luogu.org/problem/show?pid=1801
题解:
堆的巧妙应(ru)用(men)
由于所有询问第 <nobr> k </nobr>小值的k是不断递增的,所以考虑可以使用堆来维护
建立一个大根堆和一个小根堆,用大根堆来维护前k小的数,大根堆中数的个数等于就应该等于k,此时堆顶就应该是第k小的数
每加一个数时,把该数加入大根堆,如果大根堆的大小大于k,不断弹出堆顶加进小根堆
每次查询时直接访问大根堆的堆顶
如果要查询的第k小值的k增加,则将小根堆堆顶不断弹出,加入到大根堆中即可
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define reg register int
using namespace std;
int n,m;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;
int a[200010],b[200010];
int main()
{
scanf("%d%d",&n,&m);
for(reg i=1;i<=n;i++) scanf("%d",&a[i]);
for(reg i=1;i<=m;i++) scanf("%d",&b[i]);
sort(b+1,b+m+1);
int pos=1;
for(int i=1;i<=n;i++)
{
q1.push(a[i]);
if(q1.size()>pos)
{
q2.push(q1.top());
q1.pop();
}
while(b[pos]==i)
{
pos++;
printf("%d\n",q1.top());
if(!q2.empty())
{
q1.push(q2.top());
q2.pop();
}
}
}
return 0;
}