132. 小组队列
有 n 个小组要排成一个队列,每个小组中有若干人。
当一个人来到队列时,如果队列中已经有了自己小组的成员,他就直接插队排在自己小组成员的后面,否则就站在队伍的最后面。
请你编写一个程序,模拟这种小组队列。
输入格式:
输入将包含一个或多个测试用例。
对于每个测试用例,第一行输入小组数量 t。
接下来 t 行,每行输入一个小组描述,第一个数表示这个小组的人数,接下来的数表示这个小组的人的编号。
编号是 0 到 999999 范围内的整数。
一个小组最多可包含 1000 个人。
最后,命令列表如下。 有三种不同的命令:
1、ENQUEUE x -
将编号是 x 的人插入队列;
2、DEQUEUE -
让整个队列的第一个人出队;
3、STOP -
测试用例结束
每个命令占一行。
当输入用例 t=0 时,代表停止输入。
需注意:测试用例最多可包含 200000(20 万)个命令,因此小组队列的实现应该是高效的:
入队和出队都需要使用常数时间。
输出样例
对于每个测试用例,首先输出一行 Scenario #k
,其中 k 是测试用例的编号。
然后,对于每个 DEQUEUE
命令,输出出队的人的编号,每个编号占一行。
在每个测试用例(包括最后一个测试用例)输出完成后,输出一个空行。
数据范围
1≤t≤1000
输入样例:
2
3 101 102 103
3 201 202 203
ENQUEUE 101
ENQUEUE 201
ENQUEUE 102
ENQUEUE 202
ENQUEUE 103
ENQUEUE 203
DEQUEUE
DEQUEUE
DEQUEUE
DEQUEUE
DEQUEUE
DEQUEUE
STOP
2
5 259001 259002 259003 259004 259005
6 260001 260002 260003 260004 260005 260006
ENQUEUE 259001
ENQUEUE 260001
ENQUEUE 259002
ENQUEUE 259003
ENQUEUE 259004
ENQUEUE 259005
DEQUEUE
DEQUEUE
ENQUEUE 260002
ENQUEUE 260003
DEQUEUE
DEQUEUE
DEQUEUE
DEQUEUE
STOP
0
输出样例:
Scenario #1
101
102
103
201
202
203
Scenario #2
259001
259002
259003
259004
259005
260001
分析
1、每个小组是一个队列。
2、小组和小组之间是队列的关系。
每个编号怎么知道这个编号是在哪个小组的呢:可以在输入的时候就记录在一个数组vis里面。
注意:pop、push函数是void类型的,empty函数是bool类型的,front、back返回的都是指针类型的。
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int N=1010, Max=1e6+10;
int vis[Max];
int main()
{
int n=0,t;
while(cin>>t, t){ //合并成while(cin>>t,t)以最后一个为判断
queue<int> q[N]; //每个小组是一个队列
queue<int> team; //小组之间也是一个队列 (小组号作为队列元素)
cout<<"Scenario #"<<++n<<endl;
for(int i=0;i<t;i++){
int num;
cin>>num;
while(num--){
int x;
cin>>x;
vis[x] = i; //用于记录这个编号是在哪个组里面的
}
}
string op;
while(cin >> op, op != "STOP"){
if(op == "ENQUEUE"){
int x;
cin>>x;
int pos = vis[x];
if(q[pos].empty()) team.push(pos);
/*
如果对应的小组是空的,说明整个队列中是没有自己小组成员的,因为它是第一个;
先让自己的小组入队team,然后在让它入队自己小组的第一个元素;
如果对应小组不空,则说明整个队列中是有自己的小组的,则只需要入队自己的小组。
*/
q[pos].push(x);
}
else{
int pos = team.front(); //取队头的第一个小组
cout<<q[pos].front()<<endl;
q[pos].pop();
if(q[pos].empty()) team.pop();
}
}
cout<<endl;
}
return 0;
}