题解 | #蛇形矩阵#

import sys

n = int(input())
m = 1
g = 1
for i in range(n):
    m = m + i
    print(m, end=" ")
    g += 1
    k = m
    for p in range(g, n + 1):
        k = k + p
        print(k, end=" ")
    print("")

循环嵌套，分别控制横向和纵向的列差