POJ 1442 Black Box（大根堆小根堆）

题目大意

题目大意：
给定一个集合，有两种操作，插入一个数和查询集合中第i小的数，i每次查询一次前自增1，初始为0.
给出两种操作如下：( ADD GET )

分析

我们在每一次查询之前维护一个大根堆和一个小根堆，大根堆中元素个数个数为i，表示当前前i小的元素是哪些。剩下的数放在小根堆中。这样每次查询就只用返回大根堆堆顶元素了。在插入一个元素时，先放入小根堆，然后取出小根堆的最小元素和大根堆的最大元素比较看是否比大根堆堆顶元素更小，如果是就交换这两个元素。

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
using namespace std;
int M,N;
long long int A[30005];
long long int u[30005];
void Work()
{
     priority_queue<long long int, vector<long long int>, less<long long int> > Qmax;
     priority_queue<long long int,vector<long long int>,greater<long long int> > Qmin;
     int i=1;
     for(int j=1;j<=M;j++)
     {
          Qmin.push(A[j]);
          long long int t1=0,t2=0;
          if(!Qmin.empty() && !Qmax.empty())
          {
              t1=Qmin.top();
              t2=Qmax.top();
          }
          if(t1<t2)
          {
                Qmin.pop();Qmin.push(t2);
                Qmax.pop();Qmax.push(t1);
                t1=Qmin.top();
                t2=Qmax.top();
          }
          while(j==u[i])
          {
                Qmax.push(Qmin.top());
                if(Qmin.empty())return;
                Qmin.pop();
                cout<<Qmax.top()<<endl;
                i++;
                if(i==N+1)break;
          }
     }
}
int main()
{
     while(scanf("%d%d",&M,&N)!=EOF)
     {
           for(int i=1;i<=M;i++)
               cin>>A[i];
           for(int j=1;j<=N;j++)
               cin>>u[j];
            Work();
     }
}