写一段最小生成树matlab代码

以下是一个基于Prim算法的最小生成树Matlab代码示例：

function [MST, total_weight] = prim_mst(G, s)
% G: 邻接矩阵
% s: 起始节点

n = length(G); % 节点数
key = inf(1, n); % 到生成树的最短距离
visited = false(1, n); % 是否已加入生成树
parent = zeros(1, n); % 节点的父节点
key(s) = 0; % 起始节点到生成树的距离为0

for i = 1:n
    u = find(~visited, 1, 'first'); % 选择未加入生成树的最小key节点
    for j = 1:n
        if ~visited(j) && key(j) < key(u)
            u = j;
        end
    end
    
    visited(u) = true; % 加入生成树
    for v = 1:n
        if G(u, v) ~= 0 && ~visited(v) && G(u, v) < key(v)
            parent(v) = u; % 更新父节点
            key(v) = G(u, v); % 更新到生成树的最短距离
        end
    end
end

% 构建最小生成树
MST = zeros(n);
for i = 1:n
    if parent(i) ~= 0
        MST(i, parent(i)) = G(i, parent(i));
        MST(parent(i), i) = G(parent(i), i);
    end
end

total_weight = sum(sum(MST))/2; % 最小生成树的总权重
end

这段代码实现了Prim算法，其中输入参数G为邻接矩阵，s为起始节点。返回值为最小生成树MST和最小生成树的总权重total_weight。请注意，该代码并不是最优的实现方式，有很多可以优化的地方。