题意:
给你n个插入的数和m次查询,每次查询输入一个数表示在第几次插入操作以后查询(是第几次查询就输出第几个小的数)
思路:
建立一个最大堆和最小堆,把一开始插入的数都存在最小堆里,每查询一次就将最小堆的堆顶存入到最大堆里(前提是保证最小堆的堆顶比最大堆的堆顶大,否则需要交换堆顶值)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
const int maxn = 30010;
priority_queue<int, vector<int>, greater<int> > hmin; //最小堆
priority_queue<int, vector<int>, less<int> > hmax; //最大堆
int s[maxn];
int main()
{
int n,m;
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&s[i]);
int k = 1;
int x;
for(int i=1;i<=m;i++)
{
scanf("%d",&x);
while(k<=x)
{
hmin.push(s[k]);
k++;
}
while(!hmax.empty() && hmin.top()<hmax.top()) //保证小根堆的元素比大根堆的大
{
int a = hmin.top();
hmin.pop();
hmin.push(hmax.top());
hmax.pop();
hmax.push(a);
}
printf("%d\n",hmin.top());
hmax.push(hmin.top());
hmin.pop();
}
return 0;
}
参考博客:https://blog.csdn.net/lin375691011/article/details/10027439