题解 | #蛇形矩阵#

import sys

n = int(input())
for a in range(0, n):  # a为行，从0开始
    o =sum(range(0, a+1))+1
    for b in range(0, n - a):  # b为列，从0开始
        p = sum(range(2+a, a+b+2))
        print(o + p, end=" ")
    print("")

o确定首列数，p确定每列增长数