Превышен лимит времени Каттиса для алгоритма Дейкстры для задачи кратчайшего пути

Question

Я сейчас пытаюсь решить: https://open.kattis.com/problems/shortestpath1, что является проблемой кратчайшего пути с положительными весами, и я довольно плохо знаком с Python.

Я реализовал алгоритм и основную функцию,Он отлично работает для их первого тестового случая, но превышает время> 3 с для второго случая, в результате чего мой код не принимается в качестве ответа, и я считаю, что в моем коде есть ошибка (и на самом деле не оптимизацияошибка, как я полагаю, 3 секунды должно быть достаточно времени).

Что я могу сделать, чтобы улучшить мой код, чтобы он прошел второй контрольный пример?

from collections import deque, namedtuple

inf = float('inf')
Edge = namedtuple('Edge', 'start, end, cost')

def make_edge(start, end, cost=1):
    return Edge(start, end, cost)


class Graph:
def __init__(self, edges):
    self.edges = [make_edge(*edge) for edge in edges]

@property
def vertices(self):
    return set(
        sum(
            ([edge.start, edge.end] for edge in self.edges), []
        )
    )

def get_node_pairs(self, n1, n2, both_ends=True):
    if both_ends:
        node_pairs = [[n1, n2], [n2, n1]]
    else:
        node_pairs = [[n1, n2]]
    return node_pairs

def remove_edge(self, n1, n2, both_ends=True):
    node_pairs = self.get_node_pairs(n1, n2, both_ends)
    edges = self.edges[:]
    for edge in edges:
        if [edge.start, edge.end] in node_pairs:
            self.edges.remove(edge)

def add_edge(self, n1, n2, cost=1, both_ends=True):
    node_pairs = self.get_node_pairs(n1, n2, both_ends)
    #for edge in self.edges:
        #if [edge.start, edge.end] in node_pairs:
           #return ValueError('Edge {} {} already exists'.format(n1, n2))

    self.edges.append(Edge(start=n1, end=n2, cost=cost))
    if both_ends:
        self.edges.append(Edge(start=n2, end=n1, cost=cost))

@property
def neighbours(self):
    neighbours = {vertex: set() for vertex in self.vertices}
    for edge in self.edges:
        neighbours[edge.start].add((edge.end, edge.cost))

    return neighbours

def dijkstra(self, source, dest):
    #assert source in self.vertices, 'Such source node doesn\'t exist'
    distances = {vertex: inf for vertex in self.vertices}
    previous_vertices = {
        vertex: None for vertex in self.vertices
    }
    distances[source] = 0
    vertices = self.vertices.copy()

    while vertices:
        current_vertex = min(
            vertices, key=lambda vertex: distances[vertex])
        vertices.remove(current_vertex)
        if distances[current_vertex] == inf:
            break
        for neighbour, cost in self.neighbours[current_vertex]:
            alternative_route = distances[current_vertex] + cost
            if alternative_route < distances[neighbour]:
                distances[neighbour] = alternative_route
                previous_vertices[neighbour] = current_vertex

    #Deque er som list men med O(1) over O(N) for list.pop()
    path, current_vertex = deque(), dest
    while previous_vertices[current_vertex] is not None:
        path.appendleft(current_vertex)
        current_vertex = previous_vertices[current_vertex]
    if path:
        path.appendleft(current_vertex)
    #return path here to get the deque with path as list of strings
    #need the distance:
    d = 0
    for index in range(len(path)):
        for x in self.edges:
            if x.start == path[index-1] and x.end == path[index]:
                d += x.cost
    return d

if __name__ == "__main__":
    #edge1 = Edge("0", "1", 2)
    #edge2 = Edge("1", "2", 2)
    #testGraph2()
    #while dette får input..
    n, m, q, s = [0, 0, 0, 0]
    u, v, w = [0, 0, 0]
    while True:
        edges = []
        [n, m, q, s] = input().split()
        #print("n: ", n, "m: ",  m, "q", q, "s: ", s)
        if n == "0" and m == "0" and q == "0" and s == "0":
            break
        for i in range(int(m)):
            [u, v, w] = input().split()
            edges.append(Edge(u, v, int(w)))
        graph = Graph(edges)
        for j in range(int(q)):
            q = input()
            distance = graph.dijkstra(s, q)
            if distance == 0 and s != q:
                print("Impossible")
            else:
                print(distance)
        print("\n")