prim算法

原创于 2021-09-27 11:55:08 发布 · 126 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #python

数据结构专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种名为Prim的图论算法，用于在给定的带权无向图中找到一棵连通且边权之和最小的生成树。通过Python代码展示了如何使用Prim算法，并以一个具体实例4x4矩阵为例进行演示。


def prim(graph,n):
    state = [0 for i in range(n)]
    dist = [float("inf") for _ in range(n)]
    result = []

    for i in range(n):
        if dist[0] == float("inf"):
            idx = 0
            dist[0] = 0
        else:
            mx = float("inf")
            for i in range(n):
                if dist[i] < mx and not state[i]:
                    mx =dist[i]
                    idx = i
        result.append(idx)
        state[idx] = 1
        for i in range(n):
            if graph[i][idx] < dist[i] and not state[i]:
                dist[i] =  graph[i][idx]
    return result



if __name__ == '__main__':
    graph = [[1,5,3,6],
             [5,8,9,6],
             [3,9,8,6],
             [6,6,6,3]]
    print(prim(graph,4))