用python实现最小生成树--Prim算法

最新推荐文章于 2025-01-28 02:49:29 发布

heavenmark

最新推荐文章于 2025-01-28 02:49:29 发布

阅读量1w

点赞数 2

分类专栏： python 算法文章标签：算法最小生成树

本文链接：https://blog.csdn.net/heavenmark/article/details/78114735

版权

本文介绍了如何使用Matlab从Excel文件中读取邻接矩阵，并实现Prim算法来找到图的最小生成树。通过具体实例展示了算法的执行过程，包括邻接矩阵的生成、最小生成树的构建及最终结果。

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

PS:如果进来是找《用matlab画散点图，并指定点与点之间的连线》这篇文章的，请进这个链接：https://blog.csdn.net/heavenmark/article/details/82794488

一、从Excel导入邻接矩阵

import xlrd
import sys

def matrix(address):                           #读取excel生成邻接矩阵
    wb = xlrd.open_workbook(address)
    sheet1 = wb.sheet_by_name('邻接矩阵_距离')
    L = []
    for i in range(1,13):
        a = sheet1.row_values(i)
        a.remove(a[0])
        L.append([int(x) for x in a])
    # print(L)
    return L

二、实现最小生成树

import matrix
import sys

def get_tree(primgraph,chararray):
    charlist = []
    charlist.append(chararray[0])
    mid = []    #mid[i]表示生成树集合中与点i最近的点的编号
    lowcost = []    #lowcost[i]表示生成树集合中与点i最近的点构成的边最小权值 ，-1表示i已经在生成树集合中
    lowcost.append(-1)
    mid.append(0)
    n = len(chararray)
    for i in range(1,n): #初始化mid数组和lowcost数组
        low