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

分析:

这个题如果是靠单纯靠循环来解决的话是非常困难而且没有头绪的,因此可以借助偏移量来解决这个问题,可以将二维数组想象成一个网格,网格中每一个格子到其相邻格子都有一个偏移量,偏移量每次只有一个变量在变化,x或者是y,因此可以借助x,y的偏移量来解决格子在二维数组中的移动问题。

其次需要考虑的是什么时间转向的问题,因为蛇形矩阵类似与回环,走到边缘地带就要转弯,不然就像是贪吃蛇一样撞墙了,就game over了,所以撞墙是其转弯的一个限制条件。再者,对于已经赋值过位置,我们没必要再去重复赋值,所以重复是转弯的第二个条件。对于一个n*m矩阵来说,小于0,大于等于n或者大于等于m都是矩阵转弯的条件。

如何判断是否重复赋值的问题,我们知道蛇形矩阵中的每一个值都是大于0的,因此我们只需要将矩阵的初始值全部赋值为0,这样在if语句中判断要经过的地方是否值为0,就可以判断该位置是否被赋值过。

此外我还画了一个图来演示一下,因为矩阵的坐标系建立方式和笛卡尔坐标系不太一样:

java写蛇形矩阵 蛇形方阵 python_python


一目了然,点个👍吧

代码如下:

#蛇形矩阵
#利用偏移量来解决蛇形矩阵的问题
sx = [0,1,0,-1]
sy = [1,0,-1,0]
A = []
x,y,d = 0,0,0
a2 = list(map(int,input().split()))
n = a2[0]
m = a2[1]
for i in range(0,n):
    a1 = []
    for j in range(0,m):
        a1.append(0)
    A.append(a1)
for k in range(1,n * m + 1):
    A[x][y] = k
    a = x + sx[d]
    b = y + sy[d]
    if a < 0 or a >= n or b < 0 or b >= m or A[a][b] != 0:
        d = (d + 1) % 4
        a = x + sx[d]
        b = y + sy[d]
    x = a
    y = b
for i in range(0,n):
    for j in range(0,m):
        print(A[i][j],end=' ')
    print()

我觉得这个方法对我来说让我感觉清晰,如果让你明白了这个题,点个👍吧