■ 题目描述
【最小传输时延】
有M*N的节点矩阵,每个节点可以向8个方向(上、下、左、右及四个斜线方向)转发数据包,每个节点转发时会消耗固定时延,
连续两个相同时延可以减少一个时延值(即当有K个相同时延的节点连续转发时可以减少K- 1个时延值),
求左上角(0,0)开始转发数据包到右下角(M-1,N- 1)并转发出的最短时延。
输入描述
第一行两个数字,M、N,接下来有M行,每行有N个数据,表示M* N的矩阵。
输出描述
最短时延值。
示例1 输入输出示例仅供调试,后台判题数据一般不包含示例
3 3
0 2 2
1 2 1
2 2 1
输出
3
示例2 输入输出示例仅供调试,后台判题数据一般不包含示例
3 3
2 2 2
2 2 2
2 2 2
输出:
4
(2 + 2 + 2 -(3-1))
以下代码为本人原创,可以供大家参考,若有不足之处,感谢指出!!!!
m, n = map(int, input().split())
grid = []
for i in range(m):
grid.append(list(map(int, input().split())))
from heapq import *
def search(grid):
visited = set()
p = [(grid[0][0], 0, 0, grid[0][0])]
visited.add((0, 0))
heapify(p)
while p:
t, i, j, pre = heappop(p)
if i == len(grid) - 1 and j == len(grid[0]) - 1:
return t
for i_new, j_new in [(i - 1, j - 1), (i - 1, j), (i, j - 1), (i + 1, j + 1), (i + 1, j), (i, j + 1), (i - 1, j + 1), (i + 1, j - 1)]:
if 0 <= i_new < len(grid) and 0 <= j_new < len(grid[0]) and (i_new, j_new) not in visited:
visited.add((i_new, j_new))
if pre == grid[i_new][j_new]:
heappush(p, (t + grid[i_new][j_new] - 1, i_new, j_new, grid[i_new][j_new]))
else:
heappush(p, (t + grid[i_new][j_new], i_new, j_new, grid[i_new][j_new]))
out = search(grid)
print(out)