题解 | 蛇形矩阵

n = int(input())
# 计算第一列数字
frist = 1
for i in range(n):
    frist += i
    rowsum = n - i
    rowlist = []
    value = frist
    # 第一列数字不变，计算增幅得出其他列数字
    for j in range(rowsum):
        if j == 0:
            rowlist.append(value)
        else:
            value += i + j + 1
            rowlist.append(value)
    # 输出
    print(" ".join(map(str, rowlist)))

模拟