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,就可以判断该位置是否被赋值过。
此外我还画了一个图来演示一下,因为矩阵的坐标系建立方式和笛卡尔坐标系不太一样:
一目了然,点个👍吧
代码如下:
#蛇形矩阵
#利用偏移量来解决蛇形矩阵的问题
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()
我觉得这个方法对我来说让我感觉清晰,如果让你明白了这个题,点个👍吧