import logging import math logging.basicConfig(level=logging.DEBUG) logger = logging.getLogger(__name__) graph = {} graph["start"] = {} graph["start"]["a"] = 5 graph["start"]["b"] = 2 graph["a"] = {} graph["a"]["b"] = 8 graph["a"]["c"] = 4 graph["a"]["d"] = 2 graph["b"] = {} graph["b"]["a"] = 8 graph["b"]["d"] = 7 graph["c"] = {} graph["c"]["d"] = 6 graph["c"]["fin"] = 3 graph["d"] = {} graph["d"]["fin"] = 1 graph["fin"] = {} costs = {} costs["a"] = 5 costs["b"] = 2 costs["c"] = math.inf costs["d"] = math.inf costs["fin"] = math.inf parents = {} parents["a"] = "start" parents["b"] = "start" parents["c"] = None parents["d"] = None parents["fin"] = None processed = set() def find_lowest_cost_node(costs): lowest_cost = math.inf lowest_cost_node = None for node in costs: cost = costs[node] if cost < lowest_cost and node not in processed: lowest_cost = cost lowest_cost_node = node return lowest_cost_node node = find_lowest_cost_node(costs) while node is not None: cost = costs[node] neighbors = graph[node] for n in neighbors.keys(): new_cost = cost + neighbors[n] if costs[n] > new_cost: costs[n] = new_cost parents[n] = node processed.add(node) node = find_lowest_cost_node(costs) print(f"lowest cost route: {costs['fin']}") route = [] next = "fin" while next != "start": route.append(next) next = parents[next] route.append("start") print(f"route: {list(reversed(route))}")