L1-049 天梯赛座位分配（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

a=int(input())
b=[int(i) for i in input().split()]
c=sorted(b)    #将队伍数量由小到大排列
for i in range(a):
    if a==1:
        print('#1')
        start=1
        for i in range(b[0]):
            print(*range(start,start+19,2))    #所有队员隔位就坐
            start+=20
        break    #参赛的高校数为一的情况
    d_id=i+1    #该高校编号、座位分配开始位次
    u=a    #未分配高校数
    num=0    #已分配座位数
    s_id=0    #已分配完的高校数量
    d1=b[:i]    #编号位于该高校前的所有高校
    d2=b[i+1:]    #编号位于该高校后的所有高校
    jud=0    #用于调整队伍数最多的高校的位次
    print('#{}'.format(d_id))    #输出编号
    for j in range(1,b[i]+1):    #从第一次开始依次给每一所高校分配座位号
        if j>c[s_id]:    #判断是否已有高校队伍分配完座位
            if c[s_id] in d1:    #判断已分配完座位的高校队伍编号是否在该高校前
                d_id-=d1.count(c[s_id])    #若是，该高校的座位分配开始位次提前
            u-=c.count(c[s_id])    #未分配高校数减少
            s_id+=c.count(c[s_id])    #已分配的高校数量增加
        if u==1:    #判断是否只剩下一所高校、判断队伍数量最多的高校是否只有一所
            u=2    #各位就坐
            if c[-2] not in d2:    #判断最后一所高校的队伍是否会多排坐在一起
                jud=1
        print(*range(num+d_id+jud,num+d_id+jud+u*9+1,u))
        #输出，开始的座位号为已分配座位加上分配开始位次，座位号间隔为未分配高校数
        #若存在最后一所高校的队伍多排坐在一起的情况，前后排座位号要有间隔
        num+=u*10    #已分配座位数增加