Skip to content

Commit 5d2ff1a

Browse files
committed
corrected an indentation error
1 parent 38ad662 commit 5d2ff1a

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

shortest_path/dijkstra.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ def __init__(self, nodes, edges) -> None:
1111

1212
def shortest_path(self, start_node):
1313
num_node = len(self.nodes)
14-
parent = [None] * num_node # 记录路径,初始化全为None
15-
distance = [None] * num_node # 初始化最短距离全为None
14+
parent = [None] * num_node # 记录路径,初始化全为: None
15+
distance = [None] * num_node # 初始化最短距离全为: None
1616
edge_weight = defaultdict(lambda: None) # 边和权重映射
1717
node_neighbors = defaultdict(set) # 节点和其邻居映射
1818
# 初始化
@@ -25,7 +25,6 @@ def shortest_path(self, start_node):
2525
for neighbor in node_neighbors[start_node]:
2626
parent[neighbor] = start_node
2727
distance[neighbor] = edge_weight[(start_node, neighbor)]
28-
2928
not_visit = [_ for _ in range(num_node) if _ != start_node] # 还没有访问的节点
3029

3130
while len(not_visit):
@@ -111,7 +110,7 @@ def draw(DG, color_nodes, color_edges):
111110
edge_labels = nx.get_edge_attributes(DG, 'weight')
112111
# edge_labels = { (key[0],key[1]): "w:"+str(edge_labels[key]) for key in edge_labels }
113112
nx.draw_networkx_edge_labels(DG, pos, edge_labels=edge_labels) # 画权重
114-
# plt.savefig('dijkstra.png', format='PNG')
113+
plt.savefig('dijkstra.png', format='PNG')
115114
plt.show()
116115

117116

0 commit comments

Comments
 (0)