#2020.01.13训练题解#STL和并查集（H题）

题源HDU-1873

HDU-1873-看病要排队

Description
看病要排队这个是地球人都知道的常识。
不过经过细心的0068的观察，他发现了医院里排队还是有讲究的。0068所去的医院有三个医生（汗，这么少）同时看病。而看病的人病情有轻重，所以不能根据简单的先来先服务的原则。所以医院对每种病情规定了10种不同的优先级。级别为10的优先权最高，级别为1的优先权最低。医生在看病时，则会在他的队伍里面选择一个优先权最高的人进行诊治。如果遇到两个优先权一样的病人的话，则选择最早来排队的病人。
现在就请你帮助医院模拟这个看病过程。

Input
输入数据包含多组测试，请处理到文件结束。
每组数据第一行有一个正整数N(0<N<2000)表示发生事件的数目。
接下来有N行分别表示发生的事件。
一共有两种事件：
1:“IN A B”,表示有一个拥有优先级B的病人要求医生A诊治。(0<A<=3,0<B<=10)
2:“OUT A”,表示医生A进行了一次诊治，诊治完毕后，病人出院。(0<A<=3)

Output
对于每个"OUT A"事件，请在一行里面输出被诊治人的编号ID。如果该事件时无病人需要诊治，则输出"EMPTY"。
诊治人的编号ID的定义为：在一组测试中，"IN A B"事件发生第K次时，进来的病人ID即为K。从1开始编号。

Sample Input
7
IN 1 1
IN 1 2
OUT 1
OUT 2
IN 2 1
OUT 2
OUT 1
2
IN 1 1
OUT 1

Sample Output
2
EMPTY
3
1
1

题意

• 输入N代表这组测试接下来有N条数据
• “IN A B”，表示有一个拥有优先级B的病人要求医生A诊治，一共有三个医生
• “OUT A”，表示医生A进行了一次诊治，诊治完毕后，病人出院
• 优先级从1到10,1最低，优先级高的先被诊治，优先级相同先排队先被诊治
• 输出的编号即这个病人对应的IN是第几次发生的，编号不因诊治医生区分
• 输出对于OUT事件被诊治的病人编号，若无需诊治则输出EMPTY
• 详细请看中文题面，注意此题有多组输入

题解

• 用友元重载结构体运算符定义优先队列数组
• 优先队列数组第一维定长度，代表医生编号，第二维即队列实际入队出队还是可衍生的
• 然后用push、pop、empty分别代表有病人来了、被诊治完毕、判断是否已全部诊治结束
• 此优先队列的元素是结构体，包含医生编号、优先级和病人编号，相应进行输入输出即可
• 注意如果在多组输入的循环外定义优先队列，要在每组结束后进行优先队列的清空
• 详细请看代码行注释

涉及知识点

• C++的STL容器中 priority_queue 的记录、输出和函数运用
• 对于priority_queue的用法-详见链接博客介绍priority_queue
• 清空优先队列可用此循环出队→while(!que.empty()) que.pop();
• 如果是优先队列数组，可以循环数组下标，一个一个地 que[i] 进行上述操作

AC代码

#include <iostream>
#include <queue>
#include <string>
using namespace std;
struct stu
{
	int num;//医生编号
    int priority;//优先级
	int data;//病人编号
	friend bool operator < (const stu& a,const stu &b)
	{
    	if(a.priority==b.priority) return a.data > b.data;//>小的出 
    	else return a.priority < b.priority;//<大的出 
	}
}p; 
int main()
{
    int n;
    string str;
    while(cin>>n)
    {
        int k=1;//初始病人编号 
        priority_queue<stu> que[5];
		/*
			定义结构体stu类型的优先队列 
			如果定义在while循环外面则需要每次都清空队列
			注意没有clear函数，详见截图
			1.直接用空的队列对象赋值
			2.遍历出队列 
			3.使用swap，这种是最高效的，定义clear，保持STL容器的标准
		*/
        while(n--)
        {
            cin>>str;
            if(str=="IN")//有病人来了 
            {
                cin>>p.num>>p.priority;//具有优先级B的人要求医生A诊治 
                p.data=k;//录入病人编号 
				k++;
                que[p.num].push(p);//队列数组的脚标表示一个医生门前的队伍 
            }
            else//有病人被诊治结束 
            {
                int num;
                cin>>num;//输入医生编号
                if(!que[num].empty())//判断队列是否空了，来判断这个医生诊治完病人了没 
                {//没诊治完，队列非空 
                    cout<<que[num].top().data<<endl;//输出优先队列优先级最高的 
                    que[num].pop();//pop使被诊治完的病人离开队伍-出队 
                }
                else cout<<"EMPTY"<<endl;//诊治完了，队列为空 
            }
        }
    }
    return 0;
}