HDU1873 看病要排隊 STL優先隊列

題目描述

看病要排隊這個是地球人都知道的常識。
不過經過細心的0068的觀察,他發現了醫院裏排隊還是有講究的。0068所去的醫院有三個醫生(汗,這麼少)同時看病。而看病的人病情有輕重,所以不能根據簡單的先來先服務的原則。所以醫院對每種病情規定了10種不同的優先級。級別爲10的優先權最高,級別爲1的優先權最低。醫生在看病時,則會在他的隊伍裏面選擇一個優先權最高的人進行診治。如果遇到兩個優先權一樣的病人的話,則選擇最早來排隊的病人。
現在就請你幫助醫院模擬這個看病過程。

簡化題目:
三個醫生同時看病,病人按照優先級順序排隊,級別爲10的優先級最高,爲1最低。
現在有若干組入隊和出隊操作,判斷每一次出隊操作出隊了什麼編號的病人(或爲空)。

Input

輸入數據包含多組測試,請處理到文件結束。
每組數據第一行有一個正整數N(0 < N < 2000)表示發生事件的數目。
接下來有N行分別表示發生的事件。
一共有兩種事件:
1:”IN A B”,表示有一個擁有優先級B的病人要求醫生A診治.(1 <= A <= 3,1 <= B <= 10)
2:”OUT A”,表示醫生A進行了一次診治,診治完畢後,病人出院。(1 <= A <=3)

Output

對於每個”OUT A”事件,請在一行裏面輸出被診治人的編號ID。如果該事件時無病人需要診治,則輸出”EMPTY”。
診治人的編號ID的定義爲:在一組測試中,”IN A B”事件發生第K次時,進來的病人ID即爲K。從1開始編號。

思路分析

顯然隊列的出隊入隊都涉及到了優先級的問題,因此應採用優先隊列。
優先隊列的使用可以用STL的priority_queue,默認是大數在前。

注意事項

1.注意重載一個<,且重載前要加一個friend。
2.注意每次結束後把隊列內剩下的元素都彈出。

代碼

比較醜陋,沒有用優先隊列的數組,分別開了三個優先隊列:

#include<iostream>
#include<string>
#include<stdio.h>
#include<queue>
using namespace std;
struct pnt {        //patient 記錄每個病人的優先級和入隊時間 
    int imp;
    int time;
    friend bool operator < (pnt a,pnt b) {
        if(a.imp==b.imp)    return a.time>b.time;
        return a.imp<b.imp;
    }
};
/*三個優先隊列*/
priority_queue<pnt>A;
priority_queue<pnt>B;
priority_queue<pnt>C;
int main() {
    int n,a,b,k;
    string in;
    pnt tmp;
    while(scanf("%d",&n)==1) {
        k=1;
        /*將隊列裏剩餘的元素彈出*/
        while(!A.empty())   A.pop();
        while(!B.empty())   B.pop();
        while(!C.empty())   C.pop();
        for(int i=1; i<=n; i++) {
            cin>>in;
            if(in=="IN") {
                scanf("%d %d",&a,&b);
                tmp.time=k++;
                tmp.imp=b;
                if(a==1) {
                    A.push(tmp);
                } else if(a==2) {
                    B.push(tmp);
                } else {
                    C.push(tmp);
                }
            } else {
                scanf("%d",&a);
                switch(a) {
                    case 1 : {
                        if(A.empty()) { //隊列爲空則輸出EMPTY 
                            puts("EMPTY");
                        } else {        //隊列不爲空則輸出頂上的時間並出隊 
                            printf("%d\n",A.top().time);
                            A.pop();
                        }
                        break;
                    }
                    case 2: {
                        if(B.empty()) {
                            puts("EMPTY");
                        } else {
                            printf("%d\n",B.top().time);
                            B.pop();
                        }
                        break;
                    }
                    case 3: {
                        if(C.empty()) {
                            puts("EMPTY");
                        } else {
                            printf("%d\n",C.top().time);
                            C.pop();
                        }
                        break;
                    }
                }
            }
        }
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章