蓝桥杯：迷宫与陷阱 【BFS】【Python】

小明在玩一款迷宫游戏，在游戏中他要控制自己的角色离开一间由 N×N 个格子组成的 2D 迷宫。

小明的起始位置在左上角，他需要到达右下角的格子才能离开迷宫。

每一步，他可以移动到上下左右相邻的格子中（前提是目标格子可以经过）。

迷宫中有些格子小明可以经过，我们用 '.' 表示。

有些格子是墙壁，小明不能经过，我们用 '#' 表示。

此外，有些格子上有陷阱，我们用 'X' 表示。除非小明处于无敌状态，否则不能经过。

有些格子上有无敌道具，我们用 '%' 表示。

当小明第一次到达该格子时，自动获得无敌状态，无敌状态会持续 KK 步。

之后如果再次到达该格子不会获得无敌状态了。

处于无敌状态时，可以经过有陷阱的格子，但是不会拆除/毁坏陷阱，即陷阱仍会阻止没有无敌状态的角色经过。

给定迷宫，请你计算小明最少经过几步可以离开迷宫?

输入描述

输入描述

第一行包含两个整数 N,K (1≤N≤1000,1≤K≤10)。

以下 N 行包含一个N×N 的矩阵。

矩阵保证左上角和右下角是 '.'。

输出描述

一个整数表示答案。如果小明不能离开迷宫，输出 -1。

输入输出样例

示例

输入

5 3
...XX
##%#.
...#.
.###.
.....

输出

10

思路

迷宫求最短路径都需要用BFS来解决。这道题跟普通的迷宫题的区别就是多了道具以及持续时间，那只要跟解普通的迷宫题一样只需多判断一个条件 ：

1.走的那一步有没有吃到无敌道具，

2.如果吃到了就不管之前走过的路有没有走过，通通用无敌步数倒回去走

代码实现

import collections
def bfs(x,y,step,state):               #x,y表示坐标，step表示步数，state表示无敌状态的步数
    q = collections.deque()
    vis[x][y] = 1
    end_x,end_y = n-1,n-1                                       #终点坐标
    q.append((x,y,step,state))
    dir = [(-1,0),(1,0),(0,1),(0,-1)]                           #四个方向
    while q:
        curr_x,curr_y,step,state = q.popleft()
        if curr_x == end_x and curr_y == end_y:                 #如果到达终点就返回步数
            return step
        for i in dir:
            tx,ty = i
            nx,ny = curr_x + tx, curr_y + ty
            if nx >= 0 and nx < n and ny >= 0 and ny < n and mp[nx][ny] != "#":
                if not vis[nx][ny]:                             #如果没标记过
                    if mp[nx][ny] == ".":
                        vis[nx][ny] = 1
                        q.append((nx,ny,step+1,state-1 if state else 0))
                    elif mp[nx][ny] == "X" and state:
                        vis[nx][ny] = 1
                        q.append((nx,ny,step+1,state-1))
                    elif mp[nx][ny] == "%":
                        vis[nx][ny] = 1
                        q.append((nx,ny,step+1,k))
                elif (mp[nx][ny] == "." or mp[nx][ny] == "X") and state:        #如果是无敌状态，就不管之前有没有标记过，通通都走遍
                    q.append((nx,ny,step+1,state-1 if state else 0))
    return -1

n,k = map(int,input().split())
mp = [list(map(str,input())) for i in range(n)]
vis = [[0]*n for i in range(n)]
print(bfs(0,0,0,0))