[算法竞赛进阶指南] Acwing 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

思路分析

用一个队列数组存储每个队的人，再用一个队列存储每个队，入队的时候，如果队内有人就直接入队，如果队内没有人除了入队，相应的组还要入队。出队的时候，要判断一下，如果当前队内只有一个人，那么这个人出队以后，相应的组也要出队。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;

int t;
int team[maxn];

int main(){
	int cnt=0;
	while(cin>>t){
		if(t==0)
			return 0;
		cout<<"Scenario #"<<++cnt<<endl;
		for(int i=1;i<=t;i++){
			int num;
			cin>>num;
			while(num--){
				int id;
				cin>>id;
				team[id]=i;
			}
		}	
		queue<int> q[1010],team_q;
		string command;
		while(cin>>command){
			if(command=="STOP")
				break;
			if(command=="ENQUEUE"){
				int id;
				cin>>id;
				int te=team[id];
				if(!q[te].empty()){
					q[te].push(id);
				}else{
					q[te].push(id);
					team_q.push(te);
				}
			}else{
				int te=team_q.front();
				if(q[te].size()==1){
					team_q.pop();
					cout<<q[te].front()<<endl;
					q[te].pop();
				}else{
					cout<<q[te].front()<<endl;
					q[te].pop();
				}
			}
		}
		cout<<endl;
	}
	return 0;
}