用matlab语言生成一段prim算法代码

以下是一个简单的 MATLAB 实现 Prim 算法的代码：

function MST = prim_algorithm(adjacency_matrix)
% adjacency_matrix: 邻接矩阵

n = size(adjacency_matrix, 1);  % 节点数
visited = false(1, n);  % 标记节点是否已被访问
key = inf(1, n);  % 记录每个节点到生成树的最小距离
parent = zeros(1, n);  % 记录每个节点在生成树中的父节点
MST = zeros(n, n);  % 存储最小生成树的邻接矩阵

% 任选一个节点作为起点
start_node = 1;
visited(start_node) = true;
key(start_node) = 0;

% 重复 n - 1 次，每次加入一个新节点
for i = 1:n-1
    % 找到离生成树最近的节点
    min_key = inf;
    min_node = -1;
    for j = 1:n
        if ~visited(j) && key(j) < min_key
            min_key = key(j);
            min_node = j;
        end
    end
    
    % 将该节点加入生成树
    visited(min_node) = true;
    MST(parent(min_node), min_node) = adjacency_matrix(parent(min_node), min_node);
    MST(min_node, parent(min_node)) = adjacency_matrix(min_node, parent(min_node));
    
    % 更新与新加入节点相邻的节点的距离
    for j = 1:n
        if ~visited(j) && adjacency_matrix(min_node, j) < key(j)
            key(j) = adjacency_matrix(min_node, j);
            parent(j) = min_node;
        end
    end
end

其中 `adjacency_matrix` 是一个邻接矩阵，`MST` 是最小生成树的邻接矩阵。算法的核心部分是在 `for` 循环中，每次选取距离生成树最近的节点，并将其加入生成树，然后更新与新节点相邻的节点的距离。