7-13 天梯赛座位分配 (20 分)（python)

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

代码

xiaohao=int(input())#学校的个数
gx_rs=[int(i)*10 for i in input().split()]#得出["30","40","20"]，各校人数
max_rs=max(gx_rs)#最多的人数的队
zuowei=[]#每一个学生的编号
for i in range(xiaohao):
    zuowei.append([])#经过for循环最终增加三个空列表
num=0#存放编号
gx_ds=[]#各校队数
for i in gx_rs:
    gx_ds.append(i//10)
gx_ds.sort()
max_ds=max(gx_ds)
n_maxds=gx_ds.count(max_ds)
if xiaohao==1:
    rs_cha=0
else:
    rs_cha=(gx_ds[-1]-gx_ds[-2])*10
for i in range(1,max_rs+1):
    for j in range(xiaohao):
        if i>gx_rs[j]:
            continue
        if n_maxds==1 and (max_rs-i)>=rs_cha:
            num+=1
            zuowei[j].append(num)
        elif n_maxds!=1:
            num+=1
            zuowei[j].append(num)
        elif n_maxds==1 and (max_rs-i)<rs_cha:
            if num not in zuowei[j]:
                num+=1
                zuowei[j].append(num)
            elif num in zuowei[j]:
                num+=2
                zuowei[j].append(num)
if xiaohao==1:
    print("#1")
    print("1 3 5 7 9 11 13 15 17 19")
else:
    for i in range(0,xiaohao):
        print("#%d"%(i+1))
        s1=''
        for j in range(gx_rs[i]//10):
            s1=" ".join("%d"%k for k in zuowei[i][j*10:j*10+10])
            print(s1)