PTA L1-049 天梯赛座位分配（20分）（python）

更新代码

sh = int(input())
dq = [0]+list(map(int,input().split()))
flag = 0
que = sorted(dq)
z = que[-1]
z1 = 0
if sh>=2:
    z1 = que[-2]
counts = dq.count(z)
if counts==1:
    flag = 1
maxs = z*10
gest = z1*10
bianhao = [[0] for i in range(sh+1)]
hao = 0
o = 0
if sh==1 and dq[1]==1:
    print("#1")
    print("1 3 5 7 9 11 13 15 17 19")
else:
    for i in range(1,maxs+1):
        for j in range(1,sh+1):
            if i>dq[j]*10:
                continue
            hao += 1
            if flag==1 and o==j:
                hao += 1
            bianhao[j].append(hao)
            o = j
    for i in range(1,sh+1):
        print(f"#{i}")
        st = 1
        for j in range(dq[i]):
            print(" ".join(str(k) for k in bianhao[i][st:st+10]))
            st += 10

天梯赛每年有大量参赛队员，要保证同一所学校的所有队员都不能相邻，分配座位就成为一件比较麻烦的事情。为此我们制定如下策略：假设某赛场有 N 所学校参赛，第 i 所学校有 M[i] 支队伍，每队 10 位参赛选手。令每校选手排成一列纵队，第 i+1 队的选手排在第 i 队选手之后。从第 1 所学校开始，各校的第 1 位队员顺次入座，然后是各校的第 2 位队员…… 以此类推。如果最后只剩下 1 所学校的队伍还没有分配座位，则需要安排他们的队员隔位就坐。本题就要求你编写程序，自动为各校生成队员的座位号，从 1 开始编号。

注意：

4
3 3 4 3

3
3 4 2

3
3 1 1

 16分致命点，最后一个学校的剩余队伍的第一个队伍，如果是从其他学校转过来开始计算应该是加1，如果是自己学校转下来是加2.

本人是一个大一萌新，代码可能有些繁琐，还请指正，如果有不懂得还请留言，一起探讨。

输入格式：

输入在一行中给出参赛的高校数 N （不超过100的正整数）；第二行给出 N 个不超过10的正整数，其中第 i 个数对应第 i 所高校的参赛队伍数，数字间以空格分隔。

输入格式：

输入在一行中给出参赛的高校数 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())
l1 = [int(i)*10 for i in input().split()]    #l1中存放各个学校参赛人数
maxl1 = max(l1)                              #maxl1代表参赛学校当中，人数最多的学校占用的座位号个数maxl1个
l2 = []
for i in range(a):
    l2.append([])                           #l2中加入a个二维列表，用来存放座位号
num = 0             #座位号
l = []
for i in l1:                #8-17行主要是确定最后剩余学校的队伍数
    l.append(i//10)
l.sort()
s1 = max(l)
if a ==1:               #避免只有一个学校出错
    s2 = 0
else:
    s2 = l[-1]-l[-2]
for i in range(1,maxl1+1):                      #以占有座位号最多的数量循环
    for j in range(a):                      #  i代表第i个学校
        if i>l1[j]:             #判断当前学校座位号是否分配完
            continue
        if maxl1-i>=s2*10:
            num = num+1
            l2[j].append(num)
        elif maxl1-i<s2*10:             #只剩一个学校未分配座位号
            if num not in l2[j]:            #上一次分配座位号是A学校，这次分配座位号是B学校，那么这次座位号为上一次的座位号加1
                num = num+1
                l2[j].append(num)
            elif num in l2[j]:              #上一次分配座位号是A学校，这次分配座位号也是A学校，那么这次座位号为上一次座位号的加2
                num = num+2
                l2[j].append(num)
if a==1:                                #当只有一个学校时
    print("#1")
    print("1 3 5 7 9 11 13 15 17 19")
else:
    for i in range(0,a):
        print("#%d"%(i+1))
        s1 = ""
        for j in range(l1[i]//10):
            s1 = " ".join("%d"%k for k in l2[i][j*10:j*10+10])
            print(s1)