题解 | #蛇形矩阵#

import sys

for line in sys.stdin:
    n=int(line.strip())
    
    for i in range(n):
        row= i*(i+1)//2+1
        print(row, end=' ')
        for j in range(1, n-i):
            col=row+sum(range(i+2, i+2+j))
            print(col, end=' ')
        print()