POJ 1442 Black Box 优先队列

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

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

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

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

 

#include<stdio.h>
#include<cstring>
#include<iostream>
#include<map>
#include<vector>
#include<deque>
#include<queue>
using namespace std;

int a[30010],b[30010];
int main()
{
    int n, m, i, j, k, x, ans;
    while(~scanf("%d%d",&m,&n))
    {
        priority_queue<int, vector<int>, less<int> > que1;
        priority_queue<int, vector<int>, greater<int> > que2;
        for(i = 1; i <= m; i++)
            scanf("%d",&a[i]);
        for(j = 1; j <= n; j++)
            scanf("%d",&b[j]);
        i = 0;
        j = k = 1;
        while(j <= n)
        {
            if(i == b[j])  //弹出第k小的数
            {
                j++;
                if(que1.size() < k) //que1里的元素不够k个
                {
                    x = que2.top();
                    que1.push(x);
                    que2.pop();
                }
                ans = que1.top();
                printf("%d
",ans);
                k++;  //每次弹出一个数后，k的值都要加1
            }
            else
            {
                i++;
                //que1里的元素不够k个
                if(que1.size() < k)
                {
                    que2.push(a[i]);
                    x = que2.top();
                    que2.pop();
                    que1.push(x);  //先把a[i]压入que2，再从que2里取出最小值，压入que1
                }
                //如果que1的元素达到k个，且要压入队列的值比que1中的当前最大值大，说明que1中当前的最大值并不是第k小
                else if(que1.top() > a[i])
                {
                    x = que1.top();
                    que1.pop();
                    que2.push(x);
                    que1.push(a[i]);
                }
                //que1中的元素个数达到k个，且要压入队列的值比que1中的当前最大值小，说明que1中当前的最大值就是是第k小，则把a[i]直接压入que2中
                else
                {
                    que2.push(a[i]);
                }
            }
        }
    }
    return 0;
}