matlab练习程序(Prim最小生成树)

算法步骤:

1.任意找一顶点加入树中。

2.寻找所有与树相邻的元素,并取其边权重最小的并且不在树中的元素入树。

3.继续第二步,直到所有元素都入树。

效果和Kruskal算法是类似的。

matlab代码:

main.m

clear all;
close all;
clc;

G=[0 4 0 0 0 0 0 8 0;
   4 0 8 0 0 0 0 11 0;
   0 8 0 7 0 4 0 0 2;
   0 0 7 0 9 14 0 0 0;
   0 0 0 9 0 10 0 0 0;
   0 0 4 14 10 0 2 0 0;
   0 0 0 0 0 2 0 1 6;
   8 11 0 0 0 0 1 0 7;
   0 0 2 0 0 0 6 7 0];

[m n]=size(G);

q=[1];      %已经被标记的元素
k=1;        %已经标记的元素个数
A=[];       %最后产生的最小生成树
while length(q)~=m
    e=[];
    for i=1:k
        for j=1:n
            if G(q(i),j)~=0 && ~biaoji(j,q) %不在数中的元素
                e=[e;G(q(i),j) q(i) j];
            end
        end
    end
    
    [junk index]=min(e(:,1));    %求与当前标记的所有元素相邻的权重最小的边的索引
    A=[A;e(index,:)];       %最小生成树的三元组表示
    q=[q e(index,3)];
    k=k+1;  
end
clear all; close all; clc; G=[0 4 0 0 0 0 0 8 0; 4 0 8 0 0 0 0 11 0; 0 8 0 7 0 4 0 0 2; 0 0 7 0 9 14 0 0 0; 0 0 0 9 0 10 0 0 0; 0 0 4 14 10 0 2 0 0; 0 0 0 0 0 2 0 1 6; 8 11 0 0 0 0 1 0 7; 0 0 2 0 0 0 6 7 0]; [m n]=size(G); q=[1]; %已经被标记的元素 k=1; %已经标记的元素个数 A=[]; %最后产生的最小生成树 while length(q)~=m e=[]; for i=1:k for j=1:n if G(q(i),j)~=0 && ~biaoji(j,q) %不在数中的元素 e=[e;G(q(i),j) q(i) j]; end end end [junk index]=min(e(:,1)); %求与当前标记的所有元素相邻的权重最小的边的索引 A=[A;e(index,:)]; %最小生成树的三元组表示 q=[q e(index,3)]; k=k+1; end

biaoji.m

function re=biaoji(j,biao)  %判断j点是否已被标记
    l=length(biao);
    for i=1:l
       if j==biao(i) 
            re=1;
            return;
       end
    end
    re=0;
    return;
end
function re=biaoji(j,biao) %判断j点是否已被标记 l=length(biao); for i=1:l if j==biao(i) re=1; return; end end re=0; return; end
算法步骤: 1.任意找一顶点加入树中。 2.寻找所有与树相邻的元素,并取其边权重最小的并且不在树中的元素入树。 3.继续第二步,直到所有元素都入树。 效果和Kruskal算法是类似的。 matlab代码: main.m clear all; close all; clc; G=[0 4 0 0 0 0 0 8 0; 4 0 8 0 0 0 0 11 0; 0 8 0 7 0 4 0 0 2; 0 0 7 0 9 14 0 0 0; 0 0 0 9 0 10 0 0 0; 0 0 4 14 10 0 2 0 0; 0 0 0 0 0 2 0 1 6; 8 11 0 0 0 0 1 0 7; 0 0 2 0 0 0 6 7 0]; [m n]=size(G); q=[1]; %已经被标记的元素 k=1; %已经标记的元素个数 A=[]; %最后产生的最小生成树 while length(q)~=m e=[]; for i=1:k for j=1:n if G(q(i),j)~=0 && ~biaoji(j,q) %不在数中的元素 e=[e;G(q(i),j) q(i) j]; end end end [junk index]=min(e(:,1)); %求与当前标记的所有元素相邻的权重最小的边的索引 A=[A;e(index,:)]; %最小生成树的三元组表示 q=[q e(index,3)]; k=k+1; end biaoji.m function re=biaoji(j,biao) %判断j点是否已被标记 l=length(biao); for i=1:l if j==biao(i) re=1; return; end end re=0; return; end
经验分享 程序员 微信小程序 职场和发展