以codewars中3 kyu Path Finder #3: the Alpinist 为例

题干：

You are at start location [0, 0] in mountain area of NxN and you can only move in one of the four cardinal directions (i.e. North, East, South, West). Return minimal number of climb rounds to target location [N-1, N-1]. Number of climb rounds between adjacent locations is defined as difference of location altitudes (ascending or descending).

 

代码主体部分（构建图（dict），堆优化的Dijkstra算法）：

 1 import heapq
 2 
 3 
 4 def get_distance(a, b):
 5     return abs(int(a)-int(b))
 6 
 7 def get_G(area):
 8     G = {}
 9     area = area.splitlines()
10     N = len(area)
11     for i in range(N):
12         for j in range(N):
13             n = N * i + j
14             G[n] = {}
15             if i < N-1:
16                 G[n][n+N] = get_distance(area[i][j], area[i+1][j])
17             if j < N-1:
18                 G[n][n+1] = get_distance(area[i][j], area[i][j+1])
19             if i > 0:
20                 G[n][n-N] = get_distance(area[i][j], area[i-1][j])
21             if j > 0:
22                 G[n][n-1] = get_distance(area[i][j], area[i][j-1])
23     return G
24 
25 
26 def dijkstra(G, start):
27     dis = dict((key, float('inf')) for key in G)         # start 到每个点的距离
28     dis[start] = 0
29     vis = dict((key, False) for key in G)       # 是否访问过某个点
30 
31     ### 堆优化
32     pq = []
33     heapq.heappush(pq, [dis[start], start])
34 
35     path = dict((key, [start]) for key in G)     # 记录到每个点的路径
36     while len(pq) > 0:
37         v_dis, v = heapq.heappop(pq)      # 未访问的点中距离最近的点
38         if vis[v] is True:
39             continue
40         vis[v] = True
41         p = path[v].copy()   # 到v的最短路径
42         for node in G[v]:    # 与v直接相连的点
43             new_dis = dis[v] + G[v][node]     # 更新到下一个点的距离
44             if new_dis < dis[node] and (not vis[node]):   # 比较距离是否更近，是则更新相关信息
45                 dis[node] = new_dis
46                 heapq.heappush(pq, [dis[node], node])
47                 temp = p.copy()
48                 temp.append(node)
49                 path[node] = temp
50     return dis, path

测试部分：

 1 g = "\n".join([
 2   "000000",
 3   "000000",
 4   "000000",
 5   "000010",
 6   "000109",
 7   "001010"
 8 ])
 9 
10 def path_finder(area):
11     dis, path = dijkstra(get_G(g), 0)
12     print(list(dis.values())[-1])

 

其他作者解法：

 1 def path_finder(maze):
 2     grid = maze.splitlines()
 3     end = h, w = len(grid) - 1, len(grid[0]) - 1
 4     bag, seen = {(0, 0): 0}, set()
 5     while bag:
 6         x, y = min(bag, key=bag.get)
 7         rounds = bag.pop((x, y))
 8         seen.add((x, y))
 9         if (x, y) == end: return rounds
10         for u, v in (-1, 0), (0, 1), (1, 0), (0, -1):
11             m, n = x + u, y + v
12             if (m, n) in seen or not (0 <= m <= h and 0 <= n <= w): continue
13             new_rounds = rounds + abs(int(grid[x][y]) - int(grid[m][n]))
14             if new_rounds < bag.get((m, n), float('inf')): bag[m, n] = new_rounds