天梯赛L1-049 天梯赛座位分配

一、题目描述

天梯赛每年有大量参赛队员，要保证同一所学校的所有队员都不能相邻，分配座位就成为一件比较麻烦的事情。为此我们制定如下策略：假设某赛场有 N 所学校参赛，第 i 所学校有 M[i] 支队伍，每队 10 位参赛选手。令每校选手排成一列纵队，第 i+1 队的选手排在第 i 队选手之后。从第 1 所学校开始，各校的第 1 位队员顺次入座，然后是各校的第 2 位队员…… 以此类推。如果最后只剩下 1 所学校的队伍还没有分配座位，则需要安排他们的队员隔位就坐。本题就要求你编写程序，自动为各校生成队员的座位号，从 1 开始编号。

输入格式：

输入在一行中给出参赛的高校数 N （不超过100的正整数）；第二行给出 N 个不超过10的正整数，其中第 i 个数对应第 i 所高校的参赛队伍数，数字间以空格分隔。

输出格式：

从第 1 所高校的第 1 支队伍开始，顺次输出队员的座位号。每队占一行，座位号间以 1 个空格分隔，行首尾不得有多余空格。另外，每所高校的第一行按“#X”输出该校的编号X，从 1 开始。

输入样例：

3
3 4 2

 输出样例：

#1
1 4 7 10 13 16 19 22 25 28
31 34 37 40 43 46 49 52 55 58
61 63 65 67 69 71 73 75 77 79
#2
2 5 8 11 14 17 20 23 26 29
32 35 38 41 44 47 50 53 56 59
62 64 66 68 70 72 74 76 78 80
82 84 86 88 90 92 94 96 98 100
#3
3 6 9 12 15 18 21 24 27 30
33 36 39 42 45 48 51 54 57 60

二、题目分析

首先要读出题目隐藏的信息。从输出样例可以看出，题目是先给学校一排一个人，再给学校二排一个人，以此类推，直到只剩一个学校。只剩一个学校时，座位号还是不能相邻，所以每个座位之间要相隔2。可以用模拟的思想直接遍历暴力解决。

三、解题思路

我的想法是，每个学校的赛场用一个数组表示，然后开始遍历，先分给第一个学校座位号1，然后给第二个学校座位号2，把所有的n个学校都遍历过后，再回到第一个学校，分给他座位号n+1，第二个学校n+2，以次类推，就可以模拟排座位。需要注意的是，如果只有一个学校参与了比赛，这个赛场的第一个座位号是1，不能在给赛场排座位之前加1，否则这个赛场的座位就会变成2，4，6，……。而正确的应该是1，3，5，……。

四、代码

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
    int a, b, m = 0, n;
    vector<ll>p;
    cin >> n;//输入参加的学校数n
    for (int i = 0; i < n; i++) {
        cin >> b;
        p.push_back(b * 10);
        m += b * 10;
    }//输入各个学校的队伍数量,同时数组p用来记录各个学校参赛的人数，也就是队伍数乘10
    vector<vector<int>>q(101);
    ll u = 1;//u是座位号，就是最后输出的那一堆数据
    int y = n;//y表示待排列的队伍数，因为如果只剩一个学校，那这个学校的每个座位间隔都是2
    bool flag[101];//flag用来记录一个学校的人数是否排完
    for (int i = 0; i < 101; i++)flag[i] = true;//使flag的初值为真，也就是使每个学校的初始状态为“可排列”
    for (int k = 0; k < m; k++) {
        if (n != 1) {//这是有不止一个学校参赛的情况
            for (int i = 0; i < n; i++) {
                if (q[i].size() == p[i] && flag[i] == true) { flag[i] = false; y--; }//一个学校分完队员就把它忽略
                if (flag[i]) {
                    if (y == 1)u++;//如果只剩一个没分完队员的学校，那就需要每个座位号间隔是2
                    q[i].push_back(u);
                    u++;
                }}}//先给每个学校依次分一个人，分到最后一个学校再以2的间隔排座位
        else for (int i = 0; i < n; i++) {
            q[i].push_back(u);
            u += 2;
        }//如果只有一个学校参赛，那就直接一个等差数列解决了。
    }
    for (int i = 0; i < n; i++) {
        cout << "#" << i + 1 << endl;
        for (int k = 0; k < p[i]; k++) {
            if ((k + 1) % 10 != 0)cout << q[i][k] << ' ';
            else if ((k + 1) != p[i] || i != n - 1)cout << q[i][k] << endl;
            else cout << q[i][k];

        }
    }//之前把各个学校的座位号储存在了q里，按要求行首行尾没有空格输出就行了
    return 0;
}