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;
}