怎么求一个堆内的第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;
}