matlab中dijsktra函数代码

以下是使用matlab实现的Dijkstra算法的示例代码。该函数名为“dijkstra”，它接受一个邻接矩阵和源节点作为输入，并返回源节点到所有其他节点的最短路径长度和路径。

function [dist, path] = dijkstra(adj_mat, src)

% 初始化
num_nodes = size(adj_mat, 1);
dist = inf(1, num_nodes);
dist(src) = 0;
visited = false(1, num_nodes);
path = cell(1, num_nodes);

% Dijkstra算法
for i = 1:num_nodes
    % 找到未访问节点中距离源节点最近的节点
    [~, cur] = min(dist .* ~visited);
    % 如果所有节点已被访问，则退出循环
    if visited(cur)
        break;
    end
    visited(cur) = true;
    % 更新与当前节点相邻的节点的距离
    for j = 1:num_nodes
        if adj_mat(cur, j) > 0
            new_dist = dist(cur) + adj_mat(cur, j);
            if new_dist < dist(j)
                dist(j) = new_dist;
                path{j} = [path{cur}, cur];
            end
        end
    end
end

% 路径格式化
for i = 1:num_nodes
    if isempty(path{i})
        continue;
    end
    path{i} = [path{i}, i];
end

在这个代码中，我们首先通过初始化将所有节点的距离设为无限大，源节点的距离为0。然后，我们循环遍历剩余的未访问节点，每次选择距离源节点最近的节点进行访问，并更新与当前节点相邻的节点的距离。最后，我们格式化路径并返回最短路径长度和路径。