华为笔试——蛇形矩阵

描述

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。

例如，当输入5时，应该输出的三角形为：

1 3 6 10 15

2 5 9 14

4 8 13

7 12

11

输入描述：

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

输出描述：

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

示例1

输入：

4

输出：

1 3 6 10
2 5 9
4 8
7

寻找规律，化繁为简，这里其实就是设置一个元素长度为n的列表，其中每个元素都是列表，然后从0-n进行遍历，对于每个遍历的元素i来说，它从i-0遍历时，对应索引的内层列表添加对应的数。

def create(n):
    if n==1:
        return [1]
    else:
        reslst=[[] for i in range(n)]
        nownum=1
        for i in range(n):
            if i==0:
                reslst[0].append(nownum)
                nownum+=1
                continue
            else:
                line=i
                for line in range(i,-1,-1):
                    reslst[line].append(nownum)
                    nownum+=1
    return reslst

n=int(input())
res=create(n)
for lst in res:
    for num in lst:
        print(num,end=' ')
    print()