黑匣子 - 堆

黑匣子

题目描述
Black Box是一种原始的数据库。它可以储存一个整数数组，还有一个特别的变量i。最开始的时候Black Box是空的．而i等于0。这个Black Box要处理一串命令。
命令只有两种：
ADD(x):把x元素放进BlackBox;
GET:i加1，然后输出Blackhox中第i小的数。
记住：第i小的数，就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如：我们来演示一下一个有11个命令的命令串。
现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多200000个。
现在用两个整数数组来表示命令串：
1.A(1)，A(2)，…A(M)：一串将要被放进Black Box的元素。每个数都是绝对值不超过2000000000的整数，M$200000。例如上面的例子就是A=(3，1，一4，2，8，-1000，2)。
2.u(1)，u(2)，…u(N)：表示第u(j)个元素被放进了Black Box里后就出现一个GET命令。例如上面的例子中u=(l，2，6，6)。输入数据不用判错。
输入格式
第一行，两个整数，M，N。
第二行，M个整数，表示A(l)……A(M)。
第三行，N个整数，表示u(l)…u(N)。
输出格式
输出Black Box根据命令串所得出的输出串，一个数字一行。
输入输出样例
输入 #1
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
输出 #1
3
3
1
2
遇到这种数据量大，且求第k大或第k小的问题。就应该想到堆来做。因为堆排序相对于其他排序来说快的多。时间复杂度更低。但是手写堆又显的麻烦，所以这里用优先队列来做，更简洁。
这题中思想就是可以建立一个最大堆，最小堆，最大堆里·存放k-1个小的元素，把更大的元素放入最小堆，这样最小堆的堆顶元素就是第k大元素。
C++代码如下：

#include<iostream>
#include<cstdio>
#include<queue>
#include<cmath>
using namespace std;
priority_queue<int> q; 
int a[10001],b[10001],c[10001],d[10001];
int main()
{
   int n,m;
   cin >> n >> m;
   for(int i = 1; i <= n; i ++)
   {
     scanf("%d%d%d",&a[i],&b[i],&c[i]);
   }
   for(int i = 1; i <= n; i ++)
   {
    for(int j = 1; j <= m; j ++)
    {
       int k = a[i] * j * j + b[i] * j + c[i];
       if(i == 1)
        q.push(k);
       else
       {
         if(k >= q.top())
         break;
         else
         {
	    q.pop();
	    q.push(k);
          }
      } 
    }
   }
   for(int i = 1; i <= m; i ++)
   {
    d[i] = q.top();
    q.pop();
   }
   for(int i = m; i >= 1; i --)
   cout << d[i] << " ";
   return 0; 
}