POJ-1442-Black Box-优先队列

题意：给定M个数，每次可以插入序列一个数；再给N个数，表示在插入第几个数时输出一个数，第一次输出序列中最小的，第二次输出序列中第二小的……以此类推，直到输出N个数。

分析：因为输出时是按照先输出最小的，再输出第二小这样的方式输出的，相当于依次输出一个有序序列中的值。但因为这个序列不是固定不变的，而是不断的在更新，所以用数组是无法实现的。我们可以用优先队列来做。

定义两个优先队列，一个用来存储前k小的数，大数在前，小数在后；另一个优先队列第k+1小到最大的数，小数在前，大数在后。每次拿到一个数，先判断第一个优先队列中的数满不满k个，如果不满k个，则直接把这个数压入到第一个队列；如果满k个，判断这个数和第一个优先队列中的第一个数的大小：如果比第一个数大，就压入第二个优先队列；如果比第一个数小，就把第一个优先队列的队首元素弹出压入第二个队列，把这个新数压入第一个优先队列。

输出时，如果第一个优先队列里的元素个数小于k，则先把第二个优先队列里的队首元素弹出压入第一个优先队列，然后输出第一个优先队列的队首元素；如果满k个，则直接输出第一个优先队列的队首元素。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#include<algorithm>
#include<iostream>
using namespace std;
int a[35000], b[35000];
int main()
{
    int n, m;
    while(~scanf("%d%d",&n,&m))
    {
        int i;
        for(i = 1; i <= n; i++)
            scanf("%d",&a[i]);
        for(i = 1; i <= m; i++)
            scanf("%d",&b[i]);
        b[0] = 0;
        int t = 1;
        priority_queue<int,vector<int >,greater<int> > q;///从小到大
        priority_queue<int,vector<int >,less<int> > p;///从大到小
        for(i = 1; i <= m; i++)
        {
            int c = b[i]-b[i-1];
            while(c--)
            {
                if(p.size()<i)///当p中的个数小于i个的话，先去把数，压到q里，再把q中最小的压到p里
                {
                    q.push(a[t++]);
                    p.push(q.top());
                    q.pop();
                }
                else
                {
                    p.push(a[t++]);///更新p的队列
                    q.push(p.top());
                    p.pop();
                }
            }
            while(p.size()<i)
            {
                p.push(q.top());
                q.pop();
            }
            printf("%d\n",p.top());
        }
    }
    return 0;
}