prim算法求最小生成树python

Prim算法是一种用于求解最小生成树的算法。在Python中，可以使用networkx库来实现Prim算法。 下面是一个例子：

import networkx as nx

# 建立一个无向图
G = nx.Graph()

# 添加边
G.add_edge('A', 'B', weight=1)
G.add_edge('A', 'C', weight=3)
G.add_edge('B', 'C', weight=2)

# 调用prim算法计算最小生成树
T = nx.prim_mst(G)

# 输出边的集合
print(T.edges())

如上面代码，我们可以通过networkx库中的prim_mst函数来计算最小生成树，返回的是一个图对象，可以通过edges()方法查看结果。