数据结构のJavaScript实现之最短路径(Dijkstra算法和Floyd算法)

Dijkstra算法——求某一个点到其他所有点的最短路径

Dijkstra算法和最小生成树的Prim算法又有异曲同工之妙。都是要将顶点分成两坨,一坨未访问的,一坨已访问的,通过循环将未访问的一次次拉下水,变成访问过的,在这个过程中,每次都找权值最小的路径。

以A点为例:

  1. 初始化A点到所有其他点的距离dis = [0, ∞, ∞](依次代表[【AA】【AB】【AC】]);
  2. 设当前点为A,当前路径dis[0] = 0;
  3. 已访问点集为{A},未访问点集为{B,C};
  4. 从未访问点集{B,C}中找到所有由当前点A和一个未访问集中的点组成的边【AC,10】【AB,8】,
  5. 遍历所有找到的边,通过(当前路径权重+找到的边的权重)和(找的边在dis中对应的权重)的比较,取更小的权重。这里∞ > 0 + 8, ∞ > 0 + 10,更新dis = [0, 8, 10];
  6. 找出dis中的包含未访问点集中某点的最小权边;此时有三条边【AA,0】【AB,8】【AC,10】,最小权是【AA,0】但是不没有包含未访问点集中的点所以不算,这里是【AB,8】;
  7. 这时A通过边【AB】拉下水了一个新的顶点B(为什么第6歩要包含未访问点集中的点);
  8. 设当前点为B,当前路径为dis[1] = 8;
  9. 更新已访问点集为{A,B},未访问点集为{C};
  10. 从未访问点集{C}中找到所有由当前点B和一个未访问集中的点组成的边【BC,1】;
  11. 遍历所有找到的边,通过(当前路径权重+找到的边的权重)和(找的边在dis中对应的权重)的比较,取更小的权重,这里10 > 8 + 1 更新dis = [0, 8, 9];
  12. 找出dis中的包含未访问点集中某点的最小权边;此时有三条边【AA,0】【AB,8】【AC,9】,【AA,0】【AB,8】没有包含未访问点集中的点所以不算,这里是【AC,9】;
  13. 这时A通过边【AC】(实际上是【AB】+【BC】)拉下水了一个新的顶点C;
  14. 设当前点为C,当前路径为dis[2] = 9;
  15. 更新已访问点集为{A,B,C},未访问点集为{};
  16. 未访问点集为空,结束。

好了,开始程序设计

还是这张图(图片来源链接

建立图表示

// 无向图
let vertices = ['A', 'B', 'C', 'D', 'E', 'F', 'G'];
let edges = [
    [0, 1, 12],
    [0, 5, 16],
    [0, 6, 14],
    [1, 2, 10],
    [1, 5, 7],
    [2, 3, 3],
    [2, 4, 5],
    [2, 5, 6],
    [3, 4, 4],
    [4, 5, 2],
    [4, 6, 8],
    [5, 6, 9]
];

一个方法,用于找到所有(由当前点和未访问集中一点组成的边)

/**
 * @description 找到当前点在未访问点集合中的所有边
 * @param {vertex} src 源顶点
 * @param {Array<vertex>} objs 目标顶点数组
 * @param {Array<edge>} edges 所有边
 * @param {Array<vertex>} vertices 所有顶点
 * @returns {Array} edges
 */
function findEdgesIn(src, objs, edges, vertices) {
    let edgesBetweenSrcObj = [];
    for (const edge of edges) {
        srcIndex = vertices.indexOf(src);
        for (const obj of objs) {
            objIndex = vertices.indexOf(obj);
            if (edge[0] === srcIndex && edge[1] === objIndex || edge[0] === objIndex && edge[1] === srcIndex) { // 无向
            // if (edge[0] === srcIndex && edge[1] === objIndex) { // 有向
                edgesBetweenSrcObj.push(edge);
            }
        }
    }
    return edgesBetweenSrcObj;
}

男主角(dijkstra)

function dijkstra(vertices, edges, startVertex) {
    let infected = []; // 已访问点集
    let remained = vertices.slice(0); // 未访问的点集
    let shortestPath = vertices.slice(0).map(d => Infinity); // 路径数组
    shortestPath[vertices.indexOf(startVertex)] = 0; // 到自己路径为0
    let cur = startVertex; // 起始顶点
    while (remained.length !== 0) {
        infected.push(cur); // 标记当前点已访问(即顶点访问顺序)
        let curIndex = vertices.indexOf(cur); // 当前点索引
        remained.splice(remained.indexOf(cur), 1); // 从未访问中删除当前点
        let foundEdges = findEdgesIn(cur, remained, edges, vertices); // 找到所有含有当前点的边
        for (const e of foundEdges) {
            let eIndex = curIndex === e[0] ? e[1] : e[0]; // 顶点索引,无向
            // let eIndex = e[1]; // 顶点索引,有向
            if(shortestPath[eIndex] > shortestPath[curIndex] + e[2]){
                shortestPath[eIndex] = shortestPath[curIndex] + e[2]; // 更新路径
            }
        }
        let newIndex = null; // 最小边新入顶点索引
        for(let i = 0; i < shortestPath.length; i ++){
            if(remained.indexOf(vertices[i]) !== -1){
                // 从未被访问的点中找一个最短路径,返回其顶点索引
                newIndex = newIndex ? (shortestPath[newIndex] < shortestPath[i] ? newIndex : i) : i;
            }
        }
        cur = vertices[newIndex]; // 更新当前顶点
    }
    return {
        path: infected,
        distance: shortestPath
    }
}

// 打印结果
{ path: [ 'A', 'B', 'G', 'F', 'C', 'D', 'E' ],
  distance: [ 0, 12, 22, 25, 27, 16, 14 ] }

Floyd算法——求各个顶点两两之间的最短路径

感谢该文章作者的讲解,本文借用其所讲的例子,故列出原文链接,尊重版权(顺带一提,博客左下角有惊喜)。
参考文章:https://www.cnblogs.com/ECJTUACM-873284962/p/6995648.html

简单粗暴的核心3个for循环

其实核心思想是:将每条路径拆分成其他两条路径的组合,如果两条路径组合的权重更小就替代之。

细节的我不码字了,上代码以供参考。老板,上图!

表示图形

// 矩阵存储
let graphMatrix = [
    [0, 2, 6, 4],
    [Infinity, 0, 3, Infinity],
    [7, Infinity, 0, 1],
    [5, Infinity, 12, 0]
];

Floyd算法

// 最短路径弗洛伊德算法
function floyd(graphMatrix) {
    let matrixCopy = graphMatrix.slice(0);
    for (let i = 0; i < graphMatrix.length; i++) {
        // Ai矩阵
        for (let j = 0; j < graphMatrix.length; j++) {
            for (let k = 0; k < graphMatrix.length; k++) {
                matrixCopy[j][k] = Math.min(matrixCopy[j][k], matrixCopy[j][i] + matrixCopy[i][k]);
            }
        }
    }
    console.log(matrixCopy)
    return matrixCopy;
}

// 打印结果
[ [ 0, 2, 5, 4 ],
  [ 9, 0, 3, 4 ],
  [ 6, 8, 0, 1 ],
  [ 5, 7, 10, 0 ] ]

 

全部评论

相关推荐

HNU_fsq:建议直接出国,这简历太6了。自愧不如
点赞 评论 收藏
分享
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务