Prim算法

最新推荐文章于 2024-10-19 22:40:15 发布

湘溶溶

最新推荐文章于 2024-10-19 22:40:15 发布

阅读量156

点赞数

分类专栏：蓝桥杯文章标签：蓝桥杯

本文链接：https://blog.csdn.net/qq_45845375/article/details/123316535

版权

蓝桥杯专栏收录该内容

16 篇文章 0 订阅

订阅专栏

本文介绍如何使用Prim算法在给定图中找到权重最小的生成树。通过逐步选择未加入的节点中与当前树相连的最小权值边，直至所有节点都被覆盖。代码实例展示了如何在Python中实现Prim算法并应用于一个6顶点的图。

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

在这里插入图片描述

_ = float('inf')#正无穷
#负无穷的表示方法 float('_inf')
def prim(graph,n):
    dis = [0]*n#用来存放找到的较小权值列表
    pre = [0]*n#用来存放每次寻找的起点
    flag = [False]*n #标志符首先初始化为false
    flag[0]=True #已经从第一个顶点开始，所以flag[0]置为true
    k = 0 #k = 0 表示列表dis 存放的是第1个顶点A相邻的权值

    for i in range(n):
        dis[i] = graph[k][i]  #graph根据题目指定
    
    for j in range(n-1): #有n-1条边
        mini = _
        for i in range(n): #n个顶点
            if mini>dis[i] and not flag[i]:
                mini = dis[i]#找到当前顶点连接的最小权值
                k = i
        if k == 0:#不连通
            return
        flag[k] = True

        #初始点和连上的点，都可以作为起点出发寻找下一个点
        #比较这两个点到下一个点的最小权值
        for i in range(n):
            if dis[i]>graph[k][i] and not flag[i]:
                dis[i] = graph[k][i]
                pre[i] = k
    return dis,pre
if __name__ == '__main__':
    n = 6
    #有六个顶点
    graph = [
        [0,6,3,_,_,_],
        [6,0,2,5,_,_],
        [3,2,0,3,4,_],
        [_,5,3,0,2,3],
        [_,_,4,2,0,5],
        [_,_,_,3,5,0]
    ]
dis,pre = prim(graph,6)
print(dis)
print(pre)