Python数组解决问题编写程序

Question

eg:输入行数5
输出为以下内容
11
7 12
4 8 13
2 5 9 14
1 3 6 10 15
用Python数组知识编写程序

Answer 1

n = int(input())
res = [[0 for j in range(n)] for i in range(n)]
i = n-1
j = 0
a = 1
while True:
    res[i][j] = a
    if i == n-1 and j == n-1:
        break
    a += 1
    if i + 1 < n:
        i += 1
        j += 1
    else:
        i = i-j-1
        j = 0
for r in res:
    for c in r:
        print(c if c != 0 else ' ', end='')
    print()