poj-1442 Black Box

        題意:爲了方便我通過解釋輸入數據來解釋題意。輸入兩個數M表示序列長度,N表示輸入N個數表示當前序列插入了幾位,例如n=1序列裏插入了3,n=6序列裏插入了3 1 -4 2 8 -1000,對於第i個輸入的n輸入當前插入的序列中第i小的數,也就是前i小的數中最大的數。

        思路:使用兩個優先隊列,遞減隊列中存前i小的數,遞增隊列中存剩下的數。每次輸入n,就先向遞增隊列中壓入n個數,然後向遞增隊列轉入i個,輸出遞增隊列的top。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#define INF 0x3f3f3f3f
#define mod 1000000007

using namespace std;

int arr[35000];
int main()
{
    int n,m,x;
    int i,j,k;
    while(~scanf("%d%d",&n,&m))
    {
        queue<int>a;
        priority_queue<int,vector<int>,greater<int> >Q;
        priority_queue<int,vector<int>,less<int> >P;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&arr[i]);
        }
        j=1;k=0;
        for(i=1;i<=m;i++)
        {
            scanf("%d",&x);
            if(x==k) j--;                                    //k作爲檢查x是否重複輸入的工具,如果重複需要重新進入下面的循環所以j--;
            for(;j<=x;j++)
            {
                    if(x!=k)
                    Q.push(arr[j]);
                    if(P.size()<i)
                    {
                        P.push(Q.top());
                        Q.pop();
                    }
                    else  if (P.size()==i)
                    {
                        if(Q.top()<P.top())
                        {
                            Q.push(P.top());
                            P.pop();
                            P.push(Q.top());
                            Q.pop();
                        }
                    }
                if(x==j) a.push(P.top());
            }
            k=x;                                                                       //k的賦值
        }
        while(!a.empty())
        {
            printf("%d\n",a.front());
            a.pop();
        }
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章