PTA 冰火人营救行动 python

小冰人和小火人是最好的朋友，小冰人被困住了，小火人现在要去营救他。小火人的所在的位置是S，冰人所在的位置是D，我们只知道每两个相邻城市之间的距离（不唯一），我们现在要找一条S到D最短的路径，让小火人去营救小冰人。

输入格式:

输入说明：输入数据的第1行给出4个正整数N、M、S、D，其中N是城市的个数，假设城市的编号为0~(N−1)；M是公路的条数；S是小火人的位置；D是小冰人的位置。随后的M行中，每行给出一条公路的信息，分别是：城市1、城市2、公路长度，中间用空格分开，数字均为整数。输入保证解的存在。

输出格式:

最短路径的大小。

输入样例:

在这里给出一组输入。例如：

4 5 0 3
0 1 1
1 3 2
0 3 4
0 2 2
2 3 1

输出样例:

在这里给出相应的输出。例如：

3

import heapq


def dijkstra(graph, start, end):

    distances = {node: float('infinity') for node in graph}
    distances[start] = 0

    priority_queue = [(0, start)]

    while priority_queue:
        current_distance, current_node = heapq.heappop(priority_queue)