蓝桥杯每日一练——蛇形矩阵 python

题目描述
蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。
输入
本题有多组数据，每组数据由一个正整数N组成。（N不大于100）
输出
对于每一组数据，输出一个N行的蛇形矩阵。两组输出之间不要额外的空行。矩阵三角中同一行的数字用一个空格分开。行尾不要多余的空格。
样例输入
5
样例输出
1 3 6 10 15
2 5 9 14
4 8 13
7 12
11

n = int(input())
lists = [[0]*n for i in range(n)]     # 创建一个纯0的列表，用来放数据
lists1 = [[0]*n for i in range(n)]    # 创建一个空列表，方便后边数据格式转换
# print(lists)
lists[0][0] = 1                        # 第一个数先赋值
for i in range(1,n):                   # 先把每一行的第一个数算好
    lists[i][0] = lists[i-1][0] + i
for i in range(n):                     # 再通过第一个数来计算出整行
    for j in range(1,n):
        lists[i][j] = lists[i][j-1]+j+i+1
for i in range(n):                     # 接下来就是输出了
    if i != 0:                         # 因为lists[i][:0]就是没有，所以要把i=0单独拎出来
        lists1[i] = [str(j) for j in lists[i][:-i]]  # 在这里把多余的数字去掉
    else:
        lists1[i] = [str(j) for j in lists[i]] # 第一行没有多余的数字
    a = ' '.join(lists1[i])
    print(a)