题解 | #蛇形矩阵#

#输入N
N=int(input())

#初始化变量
lis1=[]
i=1

#打印第一行的数字
#规律: 每一个数字是前一个数字加a
for a in range(2,N+2):
    lis1.append(i)
    i=i+a
print(' '.join(map(str,lis1)))

#打印其他行的数字
#规律: 每行数字是上一行的数字去掉最后一个数字后+i+b
for b in range(1,N):
    lis1.pop()
    for i in range(len(lis1)):
        lis1[i]=lis1[i]+i+b
    print(' '.join(map(str,lis1)))