牛客网学习笔记(HJ43 迷宫问题)(一)

125 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目来自牛客网的华为机试题库,本题目为中等题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

解析

定义一个递归函数,按照向四个方向走分别定义。想法很简单,但是具体函数的参数要细心,很容易搞错。