题解 | 蛇形矩阵

N = int(input())
if N == 1:
    print(1)
else:
    first = 1
    for i in range(N): # 按行输出
        M = N - i # 列数
        first += i
        string = str(first) # 首列
        if M > 1:
            addnum = 2 + i
            nextnum =  first + addnum
            for j in range(1,M):
                string += " " + str(nextnum)
                addnum += 1
                nextnum += addnum
        print(string)