AcWing 小组队列
Description
有n个小组要排成一个队列,每个小组中有若干人。
当一个人来到队列时,如果队列中已经有了自己小组的成员,他就直接插队排在自己小组成员的后面,否则就站在队伍的最后面。
请你编写一个程序,模拟这种小组队列。
Input
输入将包含一个或多个测试用例。
对于每个测试用例,第一行输入小组数量t。
接下来t行,每行输入一个小组描述,第一个数表示这个小组的人数,接下来的数表示这个小组的人的编号。
编号是0到999999范围内的整数。
一个小组最多可包含1000个人。
最后,命令列表如下。 有三种不同的命令:
1、ENQUEUE x - 将编号是x的人插入队列;
2、DEQUEUE - 让整个队列的第一个人出队;
3、STOP - 测试用例结束
每个命令占一行。
当输入用例t=0时,代表停止输入。
需注意:测试用例最多可包含200000(20万)个命令,因此小组队列的实现应该是高效的:
入队和出队都需要使用常数时间。
Output
对于每个测试用例,首先输出一行“Scenario #k”,其中k是测试用例的编号。
然后,对于每个DEQUEUE命令,输出出队的人的编号,每个编号占一行。
在每个测试用例(包括最后一个测试用例)输出完成后,输出一个空行。
Data Size
- 1≤t≤1000
Sample Input
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
Sample Output
Scenario #1 101 102 103 201 202 203 Scenario #2 259001 259002 259003 259004 259005 260001
题解:
- 队列。
- 一看大概是队列相关的题。瓶颈出在“如果队列中已经有了自己小组的成员,他就直接插队排在自己小组成员的后面”这个操作。
- 但其实很容易想出只需要开N个队列和一个总队列。总队列里存的是小组之间的先后关系;N个队列分别对应N个小组,存的是小组内成员的先后关系。然后,就可以光速切题了!
#include <iostream>
#include <cstdio>
#include <queue>
#define N 1005
#define M 1000005
using namespace std;
int n, cnt;
int tem[M];
queue<int> mai, zero;
queue<int> a[N];
int read()
{
int x = 0; char c = getchar();
while(c < '0' || c > '9') c = getchar();
while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
return x;
}
int main()
{
while(scanf("%d", &n) == 1)
{
if(!n) break;
printf("Scenario #%d\n", ++cnt);
mai = zero;
for(int i = 1; i <= n; i++) a[i] = zero;
for(int i = 1; i <= n; i++)
{
int num = read();
for(int j = 1; j <= num; j++)
tem[read()] = i;
}
while(1)
{
char c[10]; scanf("%s", c);
if(c[0] == 'E')
{
int x = read();
if(!a[tem[x]].size()) mai.push(tem[x]);
a[tem[x]].push(x);
}
else if(c[0] == 'D')
{
int id = mai.front();
printf("%d\n", a[id].front());
a[id].pop();
if(!a[id].size()) mai.pop();
}
else if(c[0] == 'S') break;
}
printf("\n");
}
return 0;
}