小组队列
题目描述:
$\quad$有n个小组要排成一个队列,每个小组中有若干人。当一个人来到队列时,如果队列中已经有了自己小组的成员,他就直接插队排在自己小组成员的后面,否则就站在队伍的最后面。请你编写一个程序,模拟这种小组队列。
输入格式:
$\quad$输入将包含一个或多个测试用例。对于每个测试用例,第一行输入小组数量t。接下来t行,每行输入一个小组描述,第一个数表示这个小组的人数,接下来的数表示这个小组的人的编号。编号是0到999999范围内的整数。一个小组最多可包含1000个人。
$\quad$最后,命令列表如下。 有三种不同的命令:
- 1、ENQUEUE x - 将编号是x的人插入队列;
- 2、DEQUEUE - 让整个队列的第一个人出队;
- 3、STOP - 测试用例结束
$\quad$每个命令占一行。当输入用例t=0时,代表停止输入。需注意:测试用例最多可包含200000(20万)个命令,因此小组队列的实现应该是高效的:入队和出队都需要使用常数时间。
输出样例:
$\quad$对于每个测试用例,首先输出一行“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
思路:可以给每个小组queue<int> person[i]
一个队列记录其出队入队情况,还需要用一个队列记录小组之间的顺序queue<int> team
,用一个数组记录每个小朋友id对应的组别。
- x入队:得到x的组别tid,如果该组对应的队列不为空则之间往该队列后push(x)即可;若为空,则不仅需要往该队列后push(x)即可,还需要在队列team后加入该组
- 出队:取出队列team的队首组别tid,取出tid对应的小组队列的队首元素。若取出元素后该小组队列为空,则需要将team中删除此组。
程序(CPP):
#include <iostream>
#include <queue>
#include <cstdio>
using namespace std;
const int N = 1010, M = 1e7+10;
int id[M];
int main()
{
int t;
int cnt = 1;
while(cin >> t && t)
{
printf("Scenario #%d\n", cnt++);
for(int i = 1; i <= t; i++)
{
int num; cin >> num;
for(int j = 0; j < num; j++)
{
int x; cin >> x;
id[x] = i;
}
}
string q;
queue<int> team;
queue<int> person[N];
while(cin >> q && q[0]!='S')
{
if(q[0]=='E') // 入队操作
{
int x; cin >> x;
int tid = id[x];
if(person[tid].empty()) team.push(tid);
person[tid].push(x);
}
else // 出队操作
{
int tid = team.front();
cout << person[tid].front() << endl;
person[tid].pop();
if(person[tid].empty()) team.pop();
}
}
cout << endl; // 别忘了输出格式中每一种情况后有换行
}
}