【POJ 2259】Team Queue【队列】

题目大意：

题目链接：http://poj.org/problem?id=2259
有$t$个团队的人正在排一个长队。每次新来一个人时，如果他有队友在排队，那么新人会插队到最后一个队友的身后。如果没有任何一个队友排队，则他会被排到长队的队尾。
要求支持如下3中指令：

• $ENQUEUE$ $x$：编号为$x$的人进入长队。
• $DEQUEUE$：长队的队首出队。
• $STOP$：停止模拟

对于每个$DEQUEUE$指令，输出出队的人的编号。

---

思路：

很明显的队列。但是插队操作却有点难实现。
为了完成插队操作，就要先建$n+1$个队列。$q[0]$用来储存团队的排队情况（因为一个团队会在一起），而$q[1]$到$q[n]$则储存每个团队的人员情况。
每当来了一个人时，就判断这个人的是否有队友在队伍中，如果有，就$q[team[x]].push(x)$进入团队队列；如果没有，就先将这个团队进入总队列$q[0].push(team[x])$，再讲这个人进入团队队列$q[team[x]].push(x)$。出队时，如果出队的人$x$还有队友在队伍中，那么就直接将$x$从团队队列中退出$q[team[x]].pop()$；如果没有队友了，那么不仅要然他从团队中退出，而且还要将整个团队从总队列退出$q[0].pop()$。

---

代码：

#include <cstdio>
#include <queue>
#include <cstring>
#include <string>
#include <iostream>
using namespace std;

int n,m,team[1000001],x,sum;
string s;

int main()
{
    while (++sum)
    {
        scanf("%d",&n);
        if (!n) return 0;
        printf("Scenario #%d\n",sum);
        queue<int> q[1011];  //空间换时间
        memset(team,0,sizeof(team));
        for (int i=1;i<=n;i++)
        {
            scanf("%d",&m);
            for (int j=1;j<=m;j++)
            {
                scanf("%d",&x);
                team[x]=i;
            }
        }
        while (1)
        {
            cin>>s;
            if (s[0]=='S') break;
            if (s[0]=='E')
            {
                scanf("%d",&x);
                if (!q[team[x]].size())  //没有队友
                 q[0].push(team[x]);
                q[team[x]].push(x);
            }
            if (s[0]=='D')
            {
                printf("%d\n",q[q[0].front()].front());  //输出
                q[q[0].front()].pop();
                if (!q[q[0].front()].size()) //没有队友了
                 q[0].pop();
            }
        }
        printf("\n");
    }
}