本文已参与「新人创作礼」活动,一起开启掘金创作之路。
题目来自牛客网的华为机试题库,本题目为中等题HJ43 迷宫问题
HJ43 迷宫问题
描述
定义一个二维数组 N*M ,如 5 × 5 数组下所示:
int maze[5][5] = {
0, 1, 0, 0, 0,
0, 1, 1, 1, 0,
0, 0, 0, 0, 0,
0, 1, 1, 1, 0,
0, 0, 0, 1, 0,
};
它表示一个迷宫,其中的1表示墙壁,0表示可以走的路,只能横着走或竖着走,不能斜着走,要求编程序找出从左上角到右下角的路线。入口点为[0,0],既第一格是可以走的路。
\
数据范围:2≤n,m≤10 , 输入的内容只包含0≤val≤1
输入描述:
输入两个整数,分别表示二维数组的行数,列数。再输入相应的数组,其中的1表示墙壁,0表示可以走的路。数据保证有唯一解,不考虑有多解的情况,即迷宫只有一条通道。
输出描述:
左上角到右下角的最短路径,格式如样例所示。
示例1
输入:
5 5
0 1 0 0 0
0 1 1 1 0
0 0 0 0 0
0 1 1 1 0
0 0 0 1 0
输出:
(0,0)
(1,0)
(2,0)
(2,1)
(2,2)
(2,3)
(2,4)
(3,4)
(4,4)
示例2
输入:
5 5
0 1 0 0 0
0 1 0 1 0
0 0 0 0 1
0 1 1 1 0
0 0 0 0 0
输出:
(0,0)
(1,0)
(2,0)
(3,0)
(4,0)
(4,1)
(4,2)
(4,3)
(4,4)
说明:
注意:不能斜着走!!
代码
def walk(i,j,pos):
if j+1<n and maze[i][j+1] == 0:
if (i,j+1) not in pos:
walk(i,j+1,pos+[(i,j+1)])
if j-1 >= 0 and maze[i][j-1] == 0:
if (i,j-1) not in pos:
walk(i,j-1,pos+[(i,j-1)])
if i+1<m and maze[i+1][j] == 0:
if (i+1,j) not in pos:
walk(i+1,j,pos+[(i+1,j)])
if i-1 >= 0 and maze[i-1][j] == 0:
if (i-1,j) not in pos:
walk(i-1,j,pos+[(i-1,j)])
if (i,j) == (m-1,n-1):
for p in pos:
print('(' + str(p[0]) + ',' + str(p[1]) + ')')
while True:
try:
m, n = list(map(int, input().split()))
maze = []
for _ in range(m):
maze.append(list(map(int, input().split())))
pos = [(0,0)]
walk(0,0,pos)
except:
break
解析
定义一个递归函数,按照向四个方向走分别定义。想法很简单,但是具体函数的参数要细心,很容易搞错。