给出最小生成树matlab代码

以下是Prim算法实现最小生成树的Matlab代码：

function [MST, total_weight] = prim_mst(W)
% W: n x n 带权邻接矩阵
% MST: n x n 最小生成树的邻接矩阵
% total_weight: 最小生成树的总权值

n = size(W,1); % 顶点数
visited = false(1,n); % 是否访问过
visited(1) = true; % 从1号顶点开始构建最小生成树
d = W(1,:); % 到各个顶点的最短距离
MST = zeros(n); % 最小生成树的邻接矩阵
MST_edges = []; % 最小生成树的边集

for i = 1:n-1 % 构建n-1条边
    % 选择当前最短的边
    [min_d, u] = min(d(~visited));
    u = find(d == min_d, 1);
    visited(u) = true;
    
    % 记录边和权重
    v = find(MST(u,:)); % 找到已在MST中的顶点
    if isempty(v) % u是第一个加入MST中的顶点
        total_weight = min_d;
    else
        total_weight = total_weight + min_d;
    end
    MST_edges = [MST_edges; v u];
    
    % 更新d
    d(~visited) = min(d(~visited), W(u,~visited));
    
    % 更新MST
    MST(u,v) = min_d;
    MST(v,u) = min_d;
end

end

该代码使用Prim算法实现最小生成树，输入为带权邻接矩阵W，输出为最小生成树的邻接矩阵MST和总权值total_weight。