手写matlab的prim最小生成树(注释很详细)

来源:互联网 发布:台湾娱乐圈的现状知乎 编辑:程序博客网 时间:2024/05/17 00:12
%a是邻接矩阵这里我就随便赋值啦
%result 3*n的第一、二、三行分别表示最小生成树的边的起点、终点、权集合
clear all
clc
a=zeros(7);
a(1,2)=50;a(1,3)=60;
a(2,4)=65;a(2,5)=40;
a(3,4)=52;a(3,7)=45;
a(4,5)=50;a(4,6)=30;a(4,7)=42;
a(5,6)=70;
a=a+a';a(a==0)=inf;
result=[];
p=1;
tb=2:length(a);
while size(result,2)~=length(a)-1
    temp=a(p,tb);%表示在已经形成的树每个点到下一个没标记的点的距离,以后的程序要比较以寻求最小
    temp=temp(:);
    d=min(temp);%有点类似于贪心
    [jb,kb]=find(a(p,tb)==d,1);%找到第一个最小值即可
    j=p(jb);
    k=tb(kb);
    result = [result,[j;k;d]];
    p=[p,k];
    tb(find(tb==k))=[];
end
%其实这个最小生成树也可以求得起始点到其他点的最短距离
result
原创粉丝点击