HJ35 蛇形矩阵 【python3】

题目描述

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。
例如，当输入5时，应该输出的三角形为：
1 3 6 10 15
2 5 9 14
4 8 13
7 12
11

输出描述

输入正整数N（N不大于100）

输入描述

输出一个N行的蛇形矩阵。

代码

n=int(input())
L=[[1,],]
#初始化二维列表
for i in range(n-1):
    L.append([0])
# 蛇形矩阵-第一列
for i in range(1,n):
    L[i][0]=L[i-1][0]+i
# 规律
for i in range(n):
    for j in range(1,n-i):
       L[i].append(L[i][j-1]+i+j+1)
# 遍历
for i in range(n):
    for j in range(n-i):
        print(L[i][j],end=' ')
    print('')

代码解释

找规律补充数组。

1. 每一行的规律：对于蛇形矩阵固定的一行而言，相邻数字之差是公差为1的等差数列。且第i行的首项为i+1
2. 每一列的规律：同上，但是第i列的首项为i
3. 由上述规律，先填入第一行和第一列的数据，再按照规律补充完蛇形矩阵(二维列表)，最后按行输出