天梯赛座位分配

题目

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