例题 小组队列
有n个小组要排成一个队列,每个小组中有若干人。
当一个人来到队列时,如果队列中已经有了自己小组的成员,他就直接插队排在自己小组成员的后面,否则就站在队伍的最后面。
请你编写一个程序,模拟这种小组队列。
输入格式:
输入将包含一个或多个测试用例。
对于每个测试用例,第一行输入团队数量t。
接下来t行,每行输入一个团队描述,每个团队描述由属于团队的元素数量和元素本身组成。
元素是0到999999范围内的整数。
一个团队最多可包含1000个元素。
最后,命令列表如下。 有三种不同的命令:
1、ENQUEUE x - 将元素x输入团队队列
2、DEQUEUE - 处理第一个元素并将其从队列中删除
3、STOP - 测试用例结束
每个命令占一行。
当输入用例t=0时,代表停止输入。
需注意:测试用例最多可包含200000(20万)个命令,因此团队队列的实现应该是高效的:
元素的入队和出队都应该只占用O(1)时间。
输出样例
对于每个测试用例,首先输出一行“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> team[1000]
,用来存储各个小队列,其中里面的team[0]
队列用来存储各个小队列的顺序。
代码
//官方代码
#include <queue>
#include <cstdio>
#include <iostream>
using namespace std;
const int N = 1000000, T = 1006;
int t, f[N], id = 0;
char s[10];
queue<int> q[T];
void Team_Queue() {
q[0] = queue<int>();
for (int i = 1; i <= t; i++) {
int n;
scanf("%d", &n);
while (n--) {
int x;
scanf("%d", &x);
f[x] = i;
}
q[i] = queue<int>();
}
cout << "Scenario #" << ++id << endl;
while (scanf("%s", s) && s[0] != 'S') {
if (s[0] == 'E') {
int x;
scanf("%d", &x);
if (q[f[x]].empty()) q[0].push(f[x]);
q[f[x]].push(x);
} else {
int x = q[0].front();
printf("%d\n", q[x].front());
q[x].pop();
if (q[x].empty()) q[0].pop();
}
}
cout << endl;
}
int main() {
while (cin >> t && t) Team_Queue();
return 0;
}