我试图在Python中运行Dijkstra算法。但是,当我在Eclipse下执行此代码时,没有显示任何内容。我看到pydev了解我的所有导入,并且没有错误信息。任何人都指出我正确的方向?
class Graph(object):
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 shortest_path(graph, initial_node, goal_node):
distances, paths = dijkstra(graph, initial_node)
route = [goal_node]
while goal_node != initial_node:
route.append(paths[goal_node])
goal_node = paths[goal_node]
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 shortest_path(g, 1, 5) == [1, 3, 6, 5]
assert shortest_path(g, 5, 1) == [5, 6, 3, 1]
assert shortest_path(g, 2, 5) == [2, 3, 6, 5]
assert shortest_path(g, 1, 4) == [1, 3, 4]
答案 0 :(得分:0)
你没有打印任何东西。如果断言没有失败,它就不会显示任何内容。
而不是assert
使用print
来了解shortest_path
是否返回这些数组。 (你实际上可以知道它,因为assert
没有引起任何错误。)
答案 1 :(得分:0)
假设g是图形对象,您可以尝试通过以下方式打印节点,边缘和距离:
print g.nodes
print g.edges
print g.distances
它将帮助您理解图形结构并尝试使用print而不是assert。 在您的情况下,这三行将打印以下内容:
set([1, 2, 3, 4, 5, 6])
{1: [2, 3, 6], 2: [1, 3, 4], 3: [1, 2, 4, 6], 4: [2, 3, 5], 5: [4, 6], 6:[1, 3, 5]}
{(1, 2): 7, (3, 2): 10, (1, 3): 9, (3, 6): 2, (4, 5): 6, (6, 1): 14, (3, 1): 9, (5, 4): 6, (2, 1): 7, (6, 3): 2, (5, 6): 9, (1, 6): 14, (4, 3): 11, (4, 2): 15, (2, 3): 10, (3, 4): 11, (2, 4): 15, (6, 5): 9}
希望它有所帮助!