百练 4127 迷宫问题

求最短路问题用BFS
使用queue，到达终点即找到最短路径，即最先到达终点的路径一定是最短路径
使用字典path_dict = {}记录路径（键(当前位置):值(上一处位置)），最终通过path_dict[4, 4]一步一步地将由终点指向起点的路径存入一个list，将list逆序输出可得由[0, 0]到[4, 4]的最短路径

描述

定义一个二维数组：

int maze[5][5] = {

0, 1, 0, 0, 0,

0, 1, 0, 1, 0,

0, 0, 0, 0, 0,

0, 1, 1, 1, 0,

0, 0, 0, 1, 0,

};

它表示一个迷宫，其中的1表示墙壁，0表示可以走的路，只能横着走或竖着走，不能斜着走，要求编程序找出从左上角到右下角的最短路线。

输入

一个5 × 5的二维数组，表示一个迷宫。数据保证有唯一解。

输出

左上角到右下角的最短路径，格式如样例所示。

样例输入

0 1 0 0 0
0 1 0 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)

grid = []  # 迷宫
for _ in range(5):
    grid.append(list(input().split()))
path_dict = {}  # 字典
path_dict[0, 0] = [-1, -1]  # {(0, 0): [-1, -1]}
queue = []
queue.append([0, 0])
directions = [[-1, 0], [1, 0], [0, 1], [0, -1]]
visited = [[0 for i in range(5)] for j in range(5)]  # 0表示没走过
visited[0][0] = 1  # 从grid[0][0]起步
while len(queue):  # BFS，最先到达终点的路径一定是最短路径
    top = queue[0]  # 队首为queue[0]，当前位置
    queue.pop(0)  # remove and return item at index (default last)
    if top[0] == 4 and top[1] == 4:
        tmp_list = []
        tmp_list.append([4, 4])
        next_pos = path_dict[4, 4]
        while not next_pos[0] == -1 and not next_pos[1] == -1:
            tmp_list.append([next_pos[0], next_pos[1]])
            next_pos = path_dict[next_pos[0], next_pos[1]]
        for i in range(len(tmp_list))[::-1]:
            print('({0}, {1})'.format(tmp_list[i][0], tmp_list[i][1]))
        break
    for direction in directions:
        new_r = top[0] + direction[0]
        new_c = top[1] + direction[1]
        # 没越界、没走过、非墙壁
        if new_r >= 0 and new_r < 5 and new_c >= 0 and new_c < 5 and not visited[new_r][new_c] and grid[new_r][new_c] == '0':
            path_dict[new_r, new_c] = [top[0], top[1]]  # 键(当前位置):值(上一处位置)
            visited[new_r][new_c] = 1
            queue.append([new_r, new_c])