题解 | 蛇形矩阵

def func():
    n = int(input())

    res = 1
    for i in range(n): # 0 1 2 3
        res = res + i
        print(res, end=' ')
        resnewrow = res
        for j in range(n-i-1): # 4 3 2 1
            resnewrow = resnewrow + (j+2) + i
            if j == n-i-1-1:
                print(resnewrow, end='\n')
            else:
                print(resnewrow, end=' ')


if __name__ == "__main__":
    func()