CCF-CSP 201609-4-交通规划 (Python) 最小堆 优先队列 Dijstr 最短路径变形 满分实现

201609-4-交通规划

问题

思路

求组成最短路径的边的长度总和，

⚠️：使用了最小堆做优先队列的，==Dijstra(迪杰斯特拉)==求解最短路径问题，组成最短路径的边的长度总和即为要改造的铁路长度

利用一个最小堆来维护所有从当前路径的距离，当前路径的终点城市，当前路径的最后一条边的长度，每次从堆中取出一条最小当前路径的距离并更新访问节点集合，直到所有节点都被访问或者堆为空为止。

⚠️：和传统的Dijstra有点不一样点

• 传统Dijstra是求出其它点到起点最小的距离即可
• 这里还需要判断到统一节点两种相等路径的取舍，使得总的最小，实际上是用变型的Dijstra来求满足到起点最短路径情况下的“最小生成树”

实现

1、传统Dijstra：到起点最小的距离放入优先队列

ln + min_list[dst] < min_list[dn]

from collections import defaultdict
from heapq import heapify, heappush, heappop

n, m = map(int, input().split()) # 城市数量n，城市间铁路数量m    [1,n]

road_dict = defaultdict(list) # 城市间铁路字典 {出发城市:[(距离,终点城市),]}
for _ in range(m):
  a, b, c = map(int, input().split()) # 城市a，城市b，长度c
  road_dict[a].append((c, b))
  road_dict[b].append((c, a))

in_set = set() # 已确定最短路径的城市
pre_list = [0, 1] + [-1 for _ in range(n-1)] # 城市最近前城市 [1,n]
min_list = [