■ 题目描述
2、士兵的任务
士兵在迷言中执行任务,迷言中危机重重,他需要在在最短的时间内到达指定的位置。你可以告诉士兵他最少需要多长时间吗?
输入一个n*m的迷言中,迷言中0为路,1为墙,2为超点,3为终点,4为陷阱,6为炸弹。
士兵只能向上下左右四个方向移动,如果路径上为墙,不能移动。已知每走—步需要花费1个单位的时间,走到陷阱上需要花费3个单位的时间,走到炸弹上将会激活炸弹将炸弹上下左右的墙炸为路。
注意点:
1、炸弹只能炸設墙,不会炸掉陷阱
2、炸弹、陷阱只能发挥一次作用
3、迷宫为最大为 25 * 25
4、用例保证士兵一定有方法能到达终点
解答要求
时间限制: C/C++ 6000ms,其他语言:12000ms
内存限制: C/C++ 256MB,其他语言:512MB
输入
第一行:n和m
第二行开始: n *m的短阵
输出
最少需要的单位时间
样例1
输入
4 4
1 1 1 1
1 6 2 1
1 1 0 1
1 3 1 1
输出
3
解释:士兵在位置2,向右走到炸弹位置,解锁下面位置,然后向下走两步到达3位置
样例2
输入
8 4
1 6 2 1
1 1 0 1
1 1 0 1
1 1 0 1
1 1 0 1
1 1 0 1
1 1 0 1
1 1 3 1
输出
7
解释:士兵在位置2,向下移动7步即可到达终点。
以下代码为本人原创,可以供大家参考,若有不足之处,感谢指出!!!!
import copy
m, n = list(map(int, input().split()))
matrix = []
for i in range(m):
matrix.append(list(map(int, input().split())))
from heapq import *
direction = [(0, 1), (0, -1), (1, 0), (-1, 0)]
for i in range(m):
for j in range(n):
if matrix[i][j] == 2:
start = [i, j]
if matrix[i][j] == 3:
end = [i, j]
def ask(i, j, matrix):
matrix[i][j] = 0
for a, b in direction:
newi = i + a
newj = j + b
if 0 <= newi < m and 0 <= newj < n:
if matrix[newi][newj] == 1:
matrix[newi][newj] = 0
return matrix
def srearch(grid, start, end):
visited = set()
p = [[0, start[0], start[1], grid]]
visited.add((start[0], start[1]))
heapify(p)
while p:
t, i, j, mat = heappop(p)
if i == end[0] and j == end[1]:
return t
for a, b in direction:
newi = i + a
newj = j + b
if 0 <= newi < m and 0 <= newj < n and (newi, newj) not in visited and mat[newi][newj] != 1:
visited.add((newi, newj))
if mat[newi][newj] == 4:
new_matrix = copy.deepcopy(mat)
new_matrix[newi][newj] = 0
heappush(p, [t+3, newi, newj, new_matrix])
elif mat[newi][newj] == 6:
new1_matrix = copy.deepcopy(mat)
new1_matrix = ask(newi, newj, new1_matrix)
heappush(p, [t + 1, newi, newj, new1_matrix])
else:
heappush(p, [t + 1, newi, newj, mat])
out = srearch(matrix, start, end)
print(out)