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

来源:互联网 发布:阿里大数据应用案例 编辑:程序博客网 时间:2024/05/16 11:55
%优点对于顶点多边少的稀疏图有效
%核心算法就是通过边的权值从小到大排序然后去除环路来生成最小生成树 
%[row col val]=find(a)表示返回非零元素值的行,列,元素值
%a是邻接矩阵
clc;clear all;
a(1,[2 3])=[50 60];%这里面给出邻接矩阵的另一种输入方式
a(2,[4 5])=[65 40];
a(3,[4 7])=[52 45]; a(4,[5,6])=[50,30];
a(4,7)=42;a(5,6)=70;
[i,j,b]=find(a);
data=[i';j';b'];
index=data(1:2,:);
loop=length(a)-1;
 result=[];
 while length(result)<loop
     temp = min(data(3,:));%找到最小权值
     flag=find(data(3,:)==temp);
     flag=flag(1);%找到一个即可
     v1=index(1,flag);%v1记录的是边的起始点
     v2=index(2,flag);%v2记录的是边的终止点
     if v1~=v2%防止出现环的结构
        result = [result data(:,flag)];
     end
     index(:,flag)=[];%将找到的边删除
     data(:,flag)=[];
 end
 result
原创粉丝点击