diff options
Diffstat (limited to 'algorithms.py')
| -rw-r--r-- | algorithms.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/algorithms.py b/algorithms.py index 8a6e2d3..1e70fa1 100644 --- a/algorithms.py +++ b/algorithms.py @@ -71,15 +71,17 @@ def find_shortest_path(nodes, source_id, target_id): # where (node_0, node_1, ... node_n) is a walk to node_n # and walk_dist is the total length of the walk in meters queue = [(0, (source_id,))] + iterations = 0 visited = set() while queue: # consider an unchecked walk walk_dist, walk = heapq.heappop(queue) walk_end = walk[-1] + iterations += 1 if walk_end == target_id: # you have reached your destination - return walk + return walk, iterations if walk_end in visited: # there exists a shorter walk to walk_end continue |