题解 | #蛇形矩阵#

import sys

for line in sys.stdin:
    num_count = int(line)
    arry = []
    if num_count == 1:
        print(1)
    else:
        arry = [num_count * [0] for i in range(num_count)]

        pre_num = 0
        for i in range(num_count):
            col = i + 1
            for j in range(num_count):
                if j == 0:
                    arry[j][i] = pre_num + col
                else:
                    if col < num_count:
                        arry[j][i] =  pre_num + col
                    else:
                        arry[j][i] = 0
                    col += 1
                pre_num = arry[j][i]
            pre_num = arry[0][i]
            
    for i in arry:
        for j in i:
            if j != 0:
                print(j,end=" ")
        print("")
应憋出来的，唉 辣鸡