poj1442（Black Box）

题目地址：Black Box

题目大意：

    有一个黑盒子可以进行两种操作,一种是ADD往盒子里存数，一种是GET查找盒子里的第几个数，默认盒子里始终是从小到大排列，然后A（1）、A（2）....A（M）为要存进盒子的数，u（1）、u（2）、....u（N）的数并不是取出第几个数，而这里u（N）存的数是插入几个A（M）到盒子里去，这里有一个P(1 <= p <= N) 和u是同序列的。输出 p-minimum number from our A(1), A(2), ..., A(u(p)) sequence.  p从1开始到N ，p为多少输出第几小的数。

解题思路 ：

    首先每次循环用到sort超时。

代码：

const int M=30005;
int s2[M],s1[M],s3[M];
int main()
{
    int m,n;
    priority_queue<int >Q;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        memset(s1,0,sizeof(s1));
        memset(s2,0,sizeof(s2));
        int i,j,k;
        for(i=1;i<=m;i++)
            scanf("%d",&s1[i]);
        int d=1;
        int cnt=0;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&s2[i]);
            for(j=d;j<=s2[i];j++)
            {
                d=j+1;
                s3[++cnt]=s1[j];
            }
            sort(s3,s3+cnt+1);
            printf("%d
",s3[i]);
        }
    }
    return 0;
}

sort最坏的时间复杂度是n*n,一般是nlog(n).所以每次循环用的很费时。

而优先队列的时间复杂度最坏log（n）。

首先用优先队列分别建立一个最小堆和一个最大堆，让最大堆始终存储前p小个数，所以最大堆的堆顶就是题目要求的get。最小堆始终存储剩余的数。关键是怎么往最大、和最小堆push。当p=1时。Q2为空，直接全都push最最小堆Q1里，最后将最小堆的堆顶Q1.top（）传给最大推Q2。这时Q2里有一个数，当p=2时，需要push到Q1时每次比较如果有小于Q2.top（），然后将最大最小堆栈顶交换，最后再将最小堆的堆顶Q1.top（）传给最大推Q2，以此类推. . .这样就保证了Q2始终存的是前P小个数。

代码：

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
//#include <map>
#include <set>
using namespace std;
/***************************************/
#define ll long long
#define int64 __int64
/***************************************/
const int INF = 0x7f7f7f7f;
const double eps = 1e-8;
const double PIE=acos(-1.0);
const int d1x[]= {0,-1,0,1};
const int d1y[]= {-1,0,1,0};
const int d2x[]= {0,-1,0,1};
const int d2y[]= {1,0,-1,0};
const int fx[]= {-1,-1,-1,0,0,1,1,1};
const int fy[]= {-1,0,1,-1,1,-1,0,1};
/*vector <int>map[N];map[a].push_back(b);int len=map[v].size();*/
/***************************************/
void openfile()
{
    freopen("data.in","rb",stdin);
    freopen("data.out","wb",stdout);
}
priority_queue<int> qi1;
priority_queue<int, vector<int>, greater<int> >qi2;
/**********************华丽丽的分割线,以上为模板部分*****************/
const int M=30005;
int s1[M],s2[M];
int main()
{
    int m,n;
    priority_queue<int >Q2;
    priority_queue<int ,vector<int>,greater<int> >Q1;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        memset(s1,0,sizeof(s1));
        memset(s2,0,sizeof(s2));
        int i,j,k;
        for(i=1;i<=m;i++)
            scanf("%d",&s1[i]);
        int d=1;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&s2[i]);
            for(j=d;j<=s2[i];j++)
            {
                d=j+1;
                Q1.push(s1[j]);
                if (!Q2.empty()&&Q1.top()<Q2.top())
                {
                    int v1=Q1.top();
                    int v2=Q2.top();
                    Q1.pop();
                    Q1.push(v2);
                    Q2.pop();
                    Q2.push(v1);
                }
            }
            int v=Q1.top();
            Q2.push(v);
            Q1.pop();
            printf("%d
",Q2.top());
        }
    }
    return 0;
}