poj 1442(堆---优先队列)

来源:互联网 发布:网络蜘蛛并行策略 编辑:程序博客网 时间:2024/05/16 10:52
   使用大顶堆和小顶堆。
    其中,对于序列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;}