使用大顶堆和小顶堆。
其中,对于序列S[1..n],及表示迭代器位置的index,大顶堆维护排序后的S[1..index-1],小顶堆维护排序后的S[index..n],
其中,对于序列S[1..n],及表示迭代器位置的index,大顶堆维护排序后的S[1..index-1],小顶堆维护排序后的S[index..n],
例如S[1..n] = 1,2,3,4,5,6,7,index = 4,则大顶堆为{1,2,3},小顶堆为{4,5,6,7}为什么要这样维护呢?因为当小堆最小的元素都大于大堆最大的元素时,那么序列中排第index个数就是小堆最小的数了。
源码:
#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
int array[30001];
priority_queue<int,vector<int>,less<int> > maxHeap;//大顶堆,用less模板比较,包含在functional头文件中
priority_queue<int,vector<int>,greater<int> > minHeap;//小顶堆
int main()
{
int n, m, i;
scanf("%d%d",&n,&m);
for ( i = 0; i < n; ++i )
scanf("%d",array+i );
int oper = 0, k = 0, t1, t2;
for ( i = 0; i < m; ++i )
{
scanf("%d",&oper);
while ( k < oper )
{
minHeap.push ( array[k] );
if ( ! maxHeap.empty() && minHeap.top() < maxHeap.top() )
{
t1 = minHeap.top(); minHeap.pop();
t2 = maxHeap.top(); maxHeap.pop();
maxHeap.push(t1); minHeap.push(t2);
}
++k;
}
printf("%d\n",minHeap.top());
maxHeap.push ( minHeap.top() );
minHeap.pop();
}
return 0;
}