看病要排队 优先级队列+映射
STL :priority_queue+map
问题描述:
看病要排队这个是地球人都知道的常识。
不过经过细心的0068的观察,他发现了医院里排队还是有讲究的。0068所去的医院有三个医生(汗,这么少)同时看病。而看病的人病情有轻重,所以不能根据简单的先来先服务的原则。所以医院对每种病情规定了10种不同的优先级。级别为10的优先权最高,级别为1的优先权最低。医生在看病时,则会在他的队伍里面选择一个优先权最高的人进行诊治。如果遇到两个优先权一样的病人的话,则选择最早来排队的病人。
现在就请你帮助医院模拟这个看病过程。
输入:
输入数据包含多组测试,请处理到文件结束。
每组数据第一行有一个正整数N(0<N<2000)表示发生事件的数目。
接下来有N行分别表示发生的事件。
一共有两种事件:
1:"IN A B",表示有一个拥有优先级B的病人要求医生A诊治。(0<A<=3,0<B<=10)
2:"OUT A",表示医生A进行了一次诊治,诊治完毕后,病人出院。(0<A<=3)
输出:
对于每个"OUT A"事件,请在一行里面输出被诊治人的编号ID。如果该事件时无病人需要诊治,则输出"EMPTY"。
诊治人的编号ID的定义为:在一组测试中,"IN A B"事件发生第K次时,进来的病人ID即为K。从1开始编号。
输入样例:
7
IN 1 1
IN 1 2
OUT 1
OUT 2
IN 2 1
OUT 2
OUT 1
2
IN 1 1
OUT 1
输出样例:
2
EMPTY
3
1
1
解题思路:将病人的看病的优先级存入优先队列priority_queue中,将病人编号与病人优先级以一 一映射的关系存入map映射容器中。根据priority_queue(大根堆)的默认的大数先出队的原则,将优先级高的先出队,再根据出队的优先级x在map中用迭代器找到第一个与x相等的优先级所映射的病人编号输出(即使有相同的优先级也没有关系:将病人编号作为key,优先级作为value,用迭代器iter->first)。
AC代码(C++):
#include<bits/stdc++.h>
using namespace std;
int main(){
int n;
while(~scanf("%d",&n)){
priority_queue<int> q1,q2,q3; //分别用来存储三个医生所诊病人的优先级
map<int,int> mp1,mp2,mp3; //分别存储病人编号与其优先级的映射关系
int br=0; //记录病人编号
for(int i=1;i<n+1;i++){
string s;int doctor,pri;
cin>>s;
scanf("%d",&doctor);
if(s=="IN"&&doctor==1){
scanf("%d",&pri);
q1.push(pri);
mp1[++br]+=pri;
}
if(s=="IN"&&doctor==2){
scanf("%d",&pri);
q2.push(pri);
mp2[++br]+=pri;
}
if(s=="IN"&&doctor==3){
scanf("%d",&pri);
q3.push(pri);
mp3[++br]+=pri;
}
if(s=="OUT"&&doctor==1){
if(q1.empty()){
printf("%s\n","EMPTY");
}
else{
map<int,int>::iterator it;
for(it=mp1.begin();it!=mp1.end();it++){
if(it->second==q1.top()){
printf("%d\n",it->first);
mp1.erase(it); //输出了就立刻删除
break; //找到就退出循环
}
}
q1.pop();
}
}
if(s=="OUT"&&doctor==2){
if(q2.empty()){
printf("%s\n","EMPTY");
}
else{
map<int,int>::iterator it;
for(it=mp2.begin();it!=mp2.end();it++){
if(it->second==q2.top()){
printf("%d\n",it->first);
mp2.erase(it);
break;
}
}
q2.pop();
}
}
if(s=="OUT"&&doctor==3){
if(q3.empty()){
printf("%s\n","EMPTY");
}
else{
map<int,int>::iterator it;
for(it=mp3.begin();it!=mp3.end();it++){
if(it->second==q3.top()){
printf("%d\n",it->first);
mp3.erase(it);
break;
}
}
q3.pop();
}
}
}
}
return 0;
}
代码看起来很长,有点暴力的味道,但是条理清晰,而且本题只有三个医生,这样更不容易出错。