算法题 蛇形矩阵（Python）

题目

输入两个整数n和m，输出一个n行m列的矩阵，将数字 1 到 n*m 按照回字蛇形填充至矩阵中。

具体矩阵形式可参考样例。

输入格式

输入共一行，包含两个整数n和m。

输出格式

输出满足要求的矩阵。

矩阵占n行，每行包含m个空格隔开的整数。

数据范围

1≤n,m≤100

输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

代码

n, m = map(int, input().split())
grids = [[0] * m for _ in range(n)]
t, l, r, b = 0, 0, m - 1,  n - 1
num = 1

while True:
    for i in range(l, r + 1): # 从左到右
        grids[t][i] = num
        num += 1
    t += 1
    if t > b: break
    
    for i in range(t, b + 1): # 从上到下
        grids[i][r] = num
        num += 1
    r -= 1
    
    if r < l: break
    
    for i in range(r, l - 1, -1): # 从右到左
        grids[b][i] = num
        num += 1
    b -= 1
    if  b < t: break
    
    for i in range(b, t - 1, -1): # 从下到上
        grids[i][l] = num
        num += 1
    l += 1
    if r < l: break

for i in range(n):
    print(" ".join(str(x) for x in grids[i]))