天梯赛座位分配 java版 PTA

天梯赛每年有大量参赛队员，要保证同一所学校的所有队员都不能相邻，分配座位就成为一件比较麻烦的事情。为此我们制定如下策略：假设某赛场有 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

代码实现

import java.util.Scanner;

//这道题很奇怪，类里边的public去掉就不会超时
//天梯赛座位分配
class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n = scan.nextInt();
        //为学校队伍
        int[] arr = new int[n];
        int max = 0;
        //循环三个学校，看谁的队伍最多
        for (int i = 0; i < n; i++) {
            arr[i] = scan.nextInt();
            if (arr[i] > max) {
                max = arr[i];
            }
        }
        //第一维代表学校，第二维代表人数
        int[][] res = new int[n][105];
        int[] school = new int[n];
        //cnt 代表座位号
        int cnt = 1;
        if (n == 1) { //只有一个学校则cnt从0开始，因为下边会先加1再赋值
            cnt = 0;
        }
        //10 * max 表示其中一个学校最多的队员数量 比如4队 则有40个人
        for (int i = 0; i < 10 * max; i++) {
            //n 为学校数量
            //这个循环就是先给第一个学校的第一位赋值，然后给第二个学校的第一位赋值
            for (int j = 0; j < n; j++) {
                // 参赛选手要小于自己学校能派出的选手
                if (i < arr[j] * 10) {
                    // 检查剩余学校
                    int num = 0;
                    for (int k = 0; k < n; k++) {
                        if (school[k] == 0) {
                            num++;
                        }
                    }
                    // 只剩下一个学校
                    if (num == 1) {
                        cnt += 1;
                        res[j][i] = cnt++;
                    } else {
                        res[j][i] = cnt++;
                    }
                }else {
                    // 赋值为1代表学校已经坐满
                    school[j] = 1;
                }
            }
        }
        
        for (int i = 0; i < n; i++) {
            System.out.println("#" + (i + 1));
            //arr[i] 表示队数
            for (int j = 0; j < 10 * arr[i]; j++) {
                //一队人数满之后换行
                if (j % 10 == 0 && j != 0) {
                    System.out.println();
                }else if (j != 0) {
                    System.out.print(" ");
                }
                //第一维代表学校，第二维代表人数
                System.out.print(res[i][j]);
            }
            //一个学校打印完后换行
            System.out.println();
        }
    }
}