题解 | 蛇形矩阵

import sys

for line in sys.stdin:
    n = int(line)
    dp = [[0] * (n) for _ in range(n)]
    # print(f'dp: {dp}')
    start = 0
    for i in range(n):
        j = i
        k = 0
        while j >= 0:
            start += 1
            dp[j][k] = start
            j -= 1
            k += 1
    # print(dp)
    for i in dp:
        for j in i:
            if j != 0:
                print(j, end=" ")
            else:
                break
        print("")

思路倒是不难。

在输出卡了一会~