题解 | #蛇形矩阵#

if __name__ == '__main__':
    n = int(input())
    la = n
    col = [1 for i in range(n)]
    for i in range(n-1):
        col[i+1] = col[i] + i + 1 #计算第一列数字
    while n > 0:
        row = la - n + 1
        res = [col[row-1] for i in range(n)] #通过每行第一个数字累加计算这一行其他数字
        for i in range(n-1):
            res[i+1] = res[i] + row + i + 1
        print(*res)
        n -= 1