这题可以转化为路径问题的原因是,我们把每个格点当作一个节点,每个节点有上下左右四个路径。如果当前节点中指示的路径和我们将要移动的路径相同那么就不用加1,如果不同就要加1.(具体做法是把上下左右四个方向和下标i对应,之后比较i和当前节点中的路径方向。)之后更新队列,把距离更短的加入队列。最后根据最短路径算法就可以得到答案。
def minCost(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
r,c = len(grid),len(grid[0])
INTMAX = 0x3f3f3f3f
dist = [INTMAX]*(r*c)
dist[0] = 0
queue = [(0,0,0)]
heapq.heapify(queue)
while queue:
d,x,y = heapq.heappop(queue)
node = c*x + y #注意转化成一维坐标是横坐标乘以列数加上纵坐标
if d > dist[node]: #可以不用访问数组,直接用距离来判断。距离大的路径不用考虑
continue
for i, (nx,ny) in enumerate([(x,y+1),(x,y-1),(x+1,y),(x-1,y)]): #i+1是与后面移动方向对应的
nnode = nx*c + ny
nd = dist[node] + (1 if (i+1) != grid[x][y] else 0) #方向不一致就是+1,一致就是0
if 0<= nx < r and 0<= ny < c and nd < dist[nnode]: #需要判断一下距离大小和是否越界
dist[nnode] = nd #更新为新的小值
heapq.heappush(queue,(nd,nx,ny)) #进队列
return dist[r*c-1] #返回最后一个点