CCF-CSP 201812-4-数据中心 (Python) 最小堆 优先队列 Prim最小生成树 满分实现

201812-4-数据中心

问题

思路

求最优树结构的流水线最大耗时。

⚠️：使用了最小堆做优先队列的，==Prim(普里姆)==求解最小生成树问题，最小生成树的最大边长即为最优树结构的流水线最大耗时。

利用一个最小堆来维护所有从起始节点出发能够到达的边，每次从堆中取出一条最小边并更新访问节点集合，直到所有节点都被访问或者堆为空为止。

代码中使用了最小堆和集合两种数据结构，具体流程如下：

1. 读入图的节点数、边数以及起始节点编号；
2. 创建一个字典用于存储图中的边信息，将每条边的起点、终点和边长加入到字典中；
3. 创建一个集合用于存储已经访问过的节点，创建一个堆用于存储从起始节点可达的边；
4. 循环直到所有节点都被访问过或者堆为空：
   • 从堆中取出一条最小的边；
   • 如果该边的终点不在已访问的节点集合中，则将其加入已访问节点集合中，并更新最大边长的值；
   • 将该边终点能到达的所有边加入堆中；
5. 输出图中的最大边长。

实现

# 导入必要的库
from heapq import heapify, heappop, heappush
from collections import defaultdict

# 从用户输入中读取图的节点数、边数以及起始节点编号。
n = int(input())  # 图中的节点数
m = int(input())  # 图中的边数
root = int(input()) - 1  # 起始节点的编号

# 创建一个字典用于存储图中的边信息，将每条边的起点、终点和边长加入到字典中。
road_set = defaultdict(list)