蓝桥杯真题——迷宫与陷阱

题目描述
小明在玩一款迷宫游戏，在游戏中他要控制自己的角色离开一间由 N×N 个格子组成的 2D 迷宫。
小明的起始位置在左上角，他需要到达右下角的格子才能离开迷宫。
每一步，他可以移动到上下左右相邻的格子中（前提是目标格子可以经过）。
迷宫中有些格子小明可以经过，我们用 '.' 表示。
有些格子是墙壁，小明不能经过，我们用 '#' 表示。
此外，有些格子上有陷阱，我们用 'X' 表示。除非小明处于无敌状态，否则不能经过。
有些格子上有无敌道具，我们用 '%' 表示。
当小明第一次到达该格子时，自动获得无敌状态，无敌状态会持续 K 步。
之后如果再次到达该格子不会获得无敌状态了。
处于无敌状态时，可以经过有陷阱的格子，但是不会拆除/毁坏陷阱，即陷阱仍会阻止没有无敌状态的角色经过。
给定迷宫，请你计算小明最少经过几步可以离开迷宫?
输入描述
第一行包含两个整数 N,K(1≤N≤1000,1≤K≤10)。
以下 N 行包含一个 N×N 的矩阵。
矩阵保证左上角和右下角是 '.'。
输出描述
一个整数表示答案。如果小明不能离开迷宫，输出 -1。
输入输出样例
示例
输入

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

输出
10
运行限制
最大运行时间：3s
最大运行内存: 256M

代码实现——python

import collections


def bfs(x, y, step, state):  # x,y表示坐标，step表示步数，state表示无敌状态的步数
    m = collections.deque()
    vis[x][y] = 1
    end_x, end_y = n - 1, n - 1  # 终点坐标
    m.append((x, y, step, state))
    dir1 = [(-1, 0), (1, 0), (0, 1), (0, -1)]  # 四个方向
    while m:
        curr_x, curr_y, step, state = m.popleft()
        if curr_x == end_x and curr_y == end_y:  # 如果到达终点就返回步数
            return step
        for i in dir1:
            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
                        m.append((nx, ny, step + 1, state - 1 if state else 0))
                    # 如果格子上有陷阱，可以通过，除非小明处于无敌状态，否则不能经过。
                    elif mp[nx][ny] == "X" and state:
                        vis[nx][ny] = 1
                        m.append((nx, ny, step + 1, state - 1))
                    # 如果格子上有无敌道具，表示可以走通
                    elif mp[nx][ny] == "%":
                        # 一次到达该格子时，自动获得无敌状态，无敌状态会持续 K 步。
                        vis[nx][ny] = 1
                        m.append((nx, ny, step + 1, k))
                        # 如果是无敌状态，就不管之前有没有标记过，通通都走遍
                elif (mp[nx][ny] == "." or mp[nx][ny] == "x") and state:
                    m.append((nx, ny, step + 1, state - 1 if state else 0))
    # 可到达的遍历完了但是仍然没有到达终点，表示不可达，返回-1
    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))

运行结果如下：