美团笔试hard刷题笔记1

总结：一开始看到题很慌，感觉又是自己做不出的题了。但是还是没有放弃，试着用DP的思想理了下思路，逐步分析后发现越来越像类dij算法，这才灵感如泉涌，最终得出思路：

1. 整体是广度优先搜索的策略，每次循环使用优先级队列pq维护的当前最近的点
2. 并使用mat来记录已经遍历的结果，防止重复遍历。
3. 加入剪枝优化，一旦出现需要的结果立刻跳出循环。
4. 最终时间复杂度为nlogn量级，其中n为图中需遍历的点数，n小于等于$10^4$

n, m, xs, ys, xt, yt = map(int, input().split())
a, b = [], []
for i in range(n):
 a.append(list(map(int, input().split())))
for i in range(n):
 b.append(list(map(int, input().split())))
# 准备工作
xt, yt = xt-1, yt-1
import heapq
pq = [(0, xs-1, ys-1)]
mat = [[-1]*m for i in range(n)]

# 用于计算对于当前时间当前十字路口那个方向可以通行，
# 同时返回去另一个方向需要等待的时间
def rule(t, i, j):
 ax, bx = a[i][j], b[i][j]
 t = t % (ax + bx)
 if t < ax:
     return True, ax - t
 else:
     return False, ax + bx - t
while pq:
 min_t, i, j = heapq.heappop(pq)
 if mat[i][j] != -1:
     continue
 mat[i][j] = min_t
 if i == xt and j == yt:
     ans = min_t
     break
 direc, t_w = rule(min_t, i, j)
 for step in [1, -1]:
     ni, nj = i + step, j + step
     if 0 <= nj < m:
         heapq.heappush(pq, (min_t + 1 + (t_w if direc else 0), i, nj))
     if 0 <= ni < n:
         heapq.heappush(pq, (min_t + 1 + (0 if direc else t_w), ni, j))
print(ans)