题意:有t个团队的人正在排一个长队。每次新来一个人时,如果他有队友在排队,那么这个新人会插队到最后一个队友身后。如果没有任何一个队友排队,则他会排到长队的队尾。输入每个团队中所有队员的编号,要求支持如下3种指令(前两种指令可以穿插进行)。
ENQUEUE:编号为X的人进入长队。
DEQUEUE:长队队首出队。
STOP:停止模拟。
ENQUEUE:编号为X的人进入长队。
DEQUEUE:长队队首出队。
STOP:停止模拟。
对于每个DEQUEUE指令,输出出队的人的编号。
思路:本题有两个队列,每个团队是一个队列,而团队整体又形成一个队列。我们用map使每个人对应他的团队编号。定义每个团队的队列为q2[团队编号],所有团队整体队列为q。
①入队时,先判断他的团队是否有人在排队,即q2[团队编号]是否为空,如果为空需要把团队编号入队q。然后把他加入自己的团q2[团队编号]。
②出队时,先输出q2[q.front()]。然后把他从自己的团队(即q2[团队编号])中出队。如果出队后团队为空,那么把他们这个团队编号从整体队列q中出队。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e3 + 5;
int main()
{
int T, CASE = 1;
while (~scanf("%d", &T) && T)
{
map<int, int> team;
for(int i = 0; i < T; i++)
{
int N;
scanf("%d", &N);
for (int j = 0; j < N; j++)
{
int num;
scanf("%d", &num);
team[num] = i;
}
}
printf("Scenario #%d\n",CASE++);
queue<int> q, q2[MAXN];
string op;
while (cin >> op)
{
if (op == "STOP") break;
if (op == "ENQUEUE")
{
int num;
scanf("%d", &num);
if (q2[team[num]].empty()) q.push(team[num]);
q2[team[num]].push(num);
}
else if (op == "DEQUEUE")
{
int t = q.front();
printf("%d\n", q2[t].front());
q2[t].pop();
if(q2[t].empty()) q.pop();
}
}
printf("\n");
}
return 0;
}