常见面试算法题-士兵的任务

■ 题目描述

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)