【发布时间】:2014-02-20 17:38:27
【问题描述】:
我有这个 python 代码来计算 Dijkstra 算法。如何在终端(ubuntu)中打印最短路径?我尝试了一些打印功能,但它们引发了不同类型的异常...... 提前致谢!
class Graph(object):
"""
A simple undirected, weighted graph
"""
def __init__(self):
self.nodes = set()
self.edges = {}
self.distances = {}
def add_node(self, value):
self.nodes.add(value)
def add_edge(self, from_node, to_node, distance):
self._add_edge(from_node, to_node, distance)
self._add_edge(to_node, from_node, distance)
def _add_edge(self, from_node, to_node, distance):
self.edges.setdefault(from_node, [])
self.edges[from_node].append(to_node)
self.distances[(from_node, to_node)] = distance
def dijkstra(graph, initial_node):
visited = {initial_node: 0}
current_node = initial_node
path = {}
nodes = set(graph.nodes)
while nodes:
min_node = None
for node in nodes:
if node in visited:
if min_node is None:
min_node = node
elif visited[node] < visited[min_node]:
min_node = node
if min_node is None:
break
nodes.remove(min_node)
cur_wt = visited[min_node]
for edge in graph.edges[min_node]:
wt = cur_wt + graph.distances[(min_node, edge)]
if edge not in visited or wt < visited[edge]:
visited[edge] = wt
path[edge] = min_node
return visited, path
def route(graph, x, y):
distances, paths = dijkstra(graph, x)
route = [y]
while y != x:
route.append(paths[y])
y = paths[y]
route.reverse()
return route
if __name__ == '__main__':
g = Graph()
g.nodes = set(range(1, 7))
g.add_edge(1, 2, 7)
g.add_edge(1, 3, 9)
g.add_edge(1, 6, 14)
g.add_edge(2, 3, 10)
g.add_edge(2, 4, 15)
g.add_edge(3, 4, 11)
g.add_edge(3, 6, 2)
g.add_edge(4, 5, 6)
g.add_edge(5, 6, 9)
assert route(g, 1, 5) == [1, 3, 6, 5]
assert route(g, 5, 1) == [5, 6, 3, 1]
assert route(g, 2, 5) == [2, 3, 6, 5]
assert route(g, 1, 4) == [1, 3, 4]
【问题讨论】:
-
所以本质上你是在问如何打印一个整数列表?您能否摆脱所有其他东西并提供您不理解或无法工作的最少问题代码?例如,我不明白为什么我们需要查看 Dijkstra 代码
-
I tried some print functions but they raised a different kinds of exceptions...什么打印功能?有哪些例外? -
我必须将此代码发送给我的老师,他要求使用函数 route(x, y) 实现 Dijkstra 算法,该函数返回从节点 x 到节点 y 的最短路径。