/Python/Python dijkstra implementation stack

Python dijkstra implementation stack

nodes = ('A', 'B', 'C', 'D', 'E', 'F', 'G')
distances = {
    'B': {'A': 5, 'D': 1, 'G': 2},
    'A': {'B': 5, 'D': 3, 'E': 12, 'F' :5},
    'D': {'B': 1, 'G': 1, 'E': 1, 'A': 3},
    'G': {'B': 2, 'D': 1, 'C': 2},
    'C': {'G': 2, 'E': 1, 'F': 16},
    'E': {'A': 12, 'D': 1, 'C': 1, 'F': 2},
    'F': {'A': 5, 'E': 2, 'C': 16}}

unvisited = {node: None for node in nodes} #using None as +inf
visited = {}
current = 'B'
currentDistance = 0
unvisited[current] = currentDistance

while True:
    for neighbour, distance in distances[current].items():
        if neighbour not in unvisited: continue
        newDistance = currentDistance + distance
        if unvisited[neighbour] is None or unvisited[neighbour] > newDistance:
            unvisited[neighbour] = newDistance
    visited[current] = currentDistance
    del unvisited[current]
    if not unvisited: break
    candidates = [node for node in unvisited.items() if node[1]]
    current, currentDistance = sorted(candidates, key = lambda x: x[1])[0]

print(visited)

Source: stackoverflow.com

Similar codes

-1 vote137 views1 answer
How to create user input in python
Python
4 votes190 views1 answer
Pandas check if value in column is in a list
Python
0 vote172 views1 answer
Index in the pool python
Python
0 vote183 views2 answers
ImportError: cannot import name 'run_evaluation' from 'rasa_nlu.evaluate' (c:\users\gwinivac\.conda\envs\chatboty\lib\site-packages\rasa_nlu\evaluate.py)
Python
0 vote179 views1 answer
Pyqt5 qtextedit change color of a specific line
Python
24 votes168 views2 answers
How to get the year and month in python
Python
0 vote181 views1 answer
Concatenate data vertically python
Python
1 vote168 views1 answer
How to make a program that identifies positives and negatives in python
Python
7 votes177 views2 answers
Check if is the last element in list python
Python
0 vote169 views3 answers
Python compiler to c++
Python