沟通无限校园网-最小生成树（贪心算法-Prim-Python）

zy20221209

已于 2024-03-16 20:15:29 修改

阅读量346

点赞数 10

分类专栏：趣学算法文章标签：贪心算法 python

于 2024-03-16 20:12:05 首次发布

本文链接：https://blog.csdn.net/zy20221209/article/details/136768788

版权

趣学算法专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该篇文章介绍了如何利用Prim算法在给定的带权无向图中寻找连接大学各部门的最小生成树，以实现最低费用的网络布线。作者通过Python实现了一个普利姆函数，展示了具体步骤和计算过程。

摘要由CSDN通过智能技术生成

题目：

如何设计网络布线，将这所大学的各个单位连通起来，并且使费用最少呢？

备注：代码的无向连通带权图用的是简略图

# 引入堆
import heapq

# 普利姆函数


def prim(graph):
    # 从图中随机选择一个节点作为起始节点
    start_node = list(graph.keys())[0]

    visited = set()  # 存储已访问过的节点
    min_spanning_tree = []  # 存储最小生成树的边
    total_cost = 0  # 总花费

    # 使用优先队列来存储边和对应的权重
    edge_heap = [(0, start_node, None)]  # (权重, 目标节点, 源节点)

    while edge_heap:
        weight, current_node, parent_node = heapq.heappop(edge_heap)

        if current_node not in visited:
            visited.add(current_node)
            if parent_node is not None:
                min_spanning_tree.append((parent_node, current_node, weight))
                total_cost += weight

            for neighbor, neighbor_weight in graph[current_node].items():
                if neighbor not in visited:
                    heapq.heappush(
                        edge_heap, (neighbor_weight, neighbor, current_node))

    return min_spanning_tree, total_cost


# 用字典表示各点之间的距离
graph = {
    '1': {'2': 23, '6': 28, '7': 36},
    '2': {'1': 23, '3': 20, '7': 1},
    '3': {'2': 20, '4': 15, '7': 4},
    '4': {'3': 15, '5': 3, '7': 9},
    '5': {'4': 3, '6': 17, '7': 16},
    '6': {'1': 28, '5': 17, '7': 25},
    '7': {'1': 36, '2': 1, '3': 4, '4': 9, '5': 16, '6': 25}
}

result, cost = prim(graph)
print("Minimum Spanning Tree:")
for edge in result:
    print(edge)
print("Total Cost:", cost)