天梯赛座位分配

164 阅读3分钟

本文已参与[新人创作礼]活动,一起开启掘金创作之路.

题目

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

java代码

不得不说,个人感觉这道题很考验细心。 具体的思路已经在注释中写明。

 import java.util.Scanner;
 ​
 public class Main {
 ​
     public static void main(String[] args) {
         Scanner sc = new Scanner(System.in);
         int n = sc.nextInt();
         int[] num = new int[n];
         for (int i = 0; i < num.length; i++) {
             num[i] = sc.nextInt();
         }
 ​
         // 获取所有学校中最多的队伍数量
         int max = num[0];
         for (int i = 0; i < num.length; i++) {
             if (num[i] > max) {
                 max = num[i];
             }
         }
 ​
         // flag用来标记[i][j]位置是否有人
         boolean[][] flag = new boolean[max * 10][n];
         for (int i = 0; i < flag.length; i++) {
             for (int j = 0; j < flag[i].length; j++) {
                 if (i >= num[j] * 10) {
                     continue;
                 }
                 flag[i][j] = true;
             }
         }
 ​
         // 当flag[i][j]为true时,ids[i][j]表示[i][j]位置的学生编号
         int[][] ids = new int[max * 10][n];
         int id = 1; // 初始从1开始编号
 ​
         // 遍历所有的i和j
         for (int i = 0; i < flag.length; i++) {
             for (int j = 0; j < flag[i].length; j++) {
                 // 如果[i][j]位置有学生
                 if (flag[i][j]) {
                     // 此题最关键的地方,如果ids[i-1][j]等于id-1说明我们上一个id填入的仍然是第j列
                     // 即上一个id填入的位置与紧接着当前的位置是同一个学校的,此时只剩下该学校未被分配编号
                     // 这时id++是为了让编号再跳一个,不与上一个相差1
                     // i--是和外层循环的i抵消,防止有学生被直接跳过未被处理
                     if (i > 0 && ids[i - 1][j] == id - 1) {
                         id++;
                         i--;
                         continue;
                     }
                     ids[i][j] = id;
                     id++;
                 }
             }
         }
 ​
         // 这里需要用StringBuilder拼接字符串不能直接输出,否则会超时
         StringBuilder[] sb = new StringBuilder[n];
         for (int i = 0; i < sb.length; i++) {
             sb[i] = new StringBuilder();
         }
 ​
         // 接下来就是字符串拼接要注意的格式(如回车,空格问题),细心检查即可
         for (int j = 0; j < n; j++) {
             for (int i = 0; i < max * 10 && flag[i][j]; i++) {
                 if (i % 10 == 0) {
                     if (i != 0)
                         sb[j].append("\n");
                     sb[j].append(ids[i][j]);
                 } else {
                     sb[j].append(" " + ids[i][j]);
                 }
             }
         }
 ​
         // 输出结果
         for (int i = 0; i < sb.length; i++) {
             System.out.println("#" + (i + 1));
             System.out.println(sb[i]);
         }
     }
 }

\