队列：小组队列（插队）

100 阅读 0 评论 66 点赞

我是靠谱客的博主优秀信封，最近开发中收集的这篇文章主要介绍队列：小组队列（插队），觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

题目链接：https://www.acwing.com/problem/content/134/
题意：有n个小组要排成一个队列，每个小组中有若干人。
当一个人来到队列时，如果队列中已经有了自己小组的成员，他就直接插队排在自己小组成员的后面，否则就站在队伍的最后面。请你编写一个程序，模拟这种小组队列。
输入格式：
输入将包含一个或多个测试用例。
对于每个测试用例，第一行输入小组数量t。
接下来t行，每行输入一个小组描述，第一个数表示这个小组的人数，接下来的数表示这个小组的人的编号。
编号是0到999999范围内的整数。
一个小组最多可包含1000个人。
最后，命令列表如下。有三种不同的命令：
1、ENQUEUE x - 将编号是x的人插入队列；
2、DEQUEUE - 让整个队列的第一个人出队；
3、STOP - 测试用例结束
每个命令占一行。
当输入用例t=0时，代表停止输入。
需注意：测试用例最多可包含200000（20万）个命令，因此小组队列的实现应该是高效的：
入队和出队都需要使用常数时间。
输出样例
对于每个测试用例，首先输出一行“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

思路：我们可以开1000个队列来维护每组成员进队的顺序，再开一个队列来维护所有小组间的顺序便可。
代码实现：

#include<iostream>
#include<queue>
#include<algorithm>

using namespace std;
const int N = 1e6 + 5, M = 1010;

int n, c;
int teamid[N];//标记每个人属于的组号

int main()
{
    while(cin >> n, n){
        cout << "Scenario #" << ++ c << endl;
        for(int i = 0; i < n; i ++ ){
            int cnt;
            cin >> cnt;
            while(cnt -- ){
                int x;
                cin >> x;
                teamid[x] = i;
            }
        }

        queue<int> team;
        queue<int> man[M];
        string op;

        while(cin >> op, op != "STOP"){
            if(op == "ENQUEUE"){
                int x;
                cin >> x;
                int tid = teamid[x];
                //如果队伍中还没有同伴就把自己的组号加进去
                if(man[tid].empty())
                    team.push(tid);
                man[tid].push(x);
            }
            else{
                int tid = team.front();
                cout << man[tid].front() << endl;
                man[tid].pop();
                if(man[tid].empty()) team.pop();
            }
        }
        cout << endl;
    }
    return 0;
}