《经典图论算法》迪杰斯特拉算法(Dijkstra)

本文涉及的产品
大数据开发治理平台 DataWorks,不限时长
实时数仓Hologres,5000CU*H 100GB 3个月
实时计算 Flink 版,5000CU*H 3个月
简介: 这个是求最短路径的迪杰斯特拉算法,另外我还写了50多种《经典图论算法》,每种都使用C++和Java两种语言实现,熟练掌握之后无论是参加蓝桥杯,信奥赛,还是其他比赛,或者是面试,都能轻松应对。

摘要:
1,迪杰斯特拉算法介绍
2,迪杰斯特拉算法的代码实现
3,迪杰斯特拉算法的堆优化
4,为什么迪杰斯特拉算法不能处理带有负权边的图

1,迪杰斯特拉算法介绍
迪杰斯特拉算法(Dijkstra)也叫狄克斯特拉算法,它使用类似广度优先搜索的方法,解决从一个顶点到其他所有顶点的最短路径问题,它解决的是加权图(不能有负权)的最短路径问题。

从起始点开始,采用贪心算法的策略,每次选择一个没被标记且距离起始点最近的顶点,把它标记下,然后更新和它邻接的顶点 ……,直到所有顶点都计算完为止。

image.png

如上图所示,假如计算从上海到其他所有城市的最短时间,上面的时间有可能是开车,有可能是高铁也可能是坐飞机,和真实距离不成正比。

我们从起始点开始,使用一个数组 dis ,数组中 dis[j] 的值表示从起始点到顶点 j 的时间,刚开始的时候,起始点到他自己为 0 ,到其他顶点都为无穷大,如下图所示。

image.png

如果想要减少从起始点到 j 的时间,唯一的方式就是需要寻找一个中转站 k 。从起始点到 k 的时间为 dis[k] ,从 k 到 j 的时间为 g[k][j] ,然后判断中转的总时间 dis[k] + g[k][j] 是否小于 dis[j] ,如果中转时间小于 dis[j] ,就更新 dis[j] 。

比如最上面图中,从起始点到南京的时间是 3 小时,如果通过杭州中转,时间就会变成 2 小时。核心代码是下面这行。

dis[j] = min(dis[j], dis[k] + g[k][j]);

迪杰斯特拉算法的解题思路如下:
1,从起始点开始计算所有和它相连的点(也就是起始点指向的点),计算完之后把起始点标记下(表示已经计算过了)。
2,找出离起始点最近且没有被标记过的点 v ,计算所有和 v 相连且没有被标记过的点,计算完之后把 v 标记下。
3,重复上面的步骤 2 ,直到所有顶点都标记完为止。

image.png
image.png
image.png

2,迪杰斯特拉算法的代码实现
迪杰斯特拉算法使用的是贪心的策略,每次都是从未标记的顶点中找到一个离起始点最近的点,用它来更新所有和它连接且未被标记过的点,代码比较简单,我们来看下。

Java 代码:

private void test() {
   
   
    int[][] g = {
   
   {
   
   0, 1, 3, 0, 0, 0},// 图的邻接矩阵。
            {
   
   0, 0, 1, 4, 2, 0},
            {
   
   0, 0, 0, 5, 5, 0},
            {
   
   0, 0, 0, 0, 0, 3},
            {
   
   0, 0, 0, 1, 0, 6},
            {
   
   0, 0, 0, 0, 0, 0}};
    dijkstra(g, 0);
}

/**
 * @param g     图的邻接矩阵
 * @param start 起始点
 */
public void dijkstra(int[][] g, int start) {
   
   
    int n = g.length;// 顶点的个数。
    int[] dis = new int[n];// 每个点到起始点的距离
    // 起始点到其他所有顶点默认给一个非常大的值,
    // 要注意下面加的时候防止出现溢出。
    Arrays.fill(dis, Integer.MAX_VALUE >> 1);
    dis[start] = 0;// 起始点到自己的值是 0 。
    boolean visited[] = new boolean[n];// 标记哪些顶点被访问过
    for (int i = 0; i < n; i++) {
   
   
        int k = -1;// 下一个没被标记且离起始点最近的顶点。
        int min = Integer.MAX_VALUE; // min 是 k 到起始点的值。
        for (int j = 0; j < n; j++) {
   
   // 寻找 k。
            if (!visited[j] && dis[j] < min) {
   
   
                min = dis[j];
                k = j;
            }
        }
        visited[k] = true;// 标记
        for (int j = 0; j < n; j++) {
   
   // 核心代码。
            // 顶点 j 没有被标记,并且 k 有到 j 的路径,并且这个路径更近,就更新。
            if (!visited[j] && g[k][j] != 0 && dis[k] + g[k][j] < dis[j])
                dis[j] = dis[k] + g[k][j];
        }
    }
    // 打印数组dis的值,测试使用。
    for (int di : dis)
        System.out.print(di + ",");
}

C++ 代码:

/**
 * @param g       图的邻接矩阵
 * @param start   起始点
 */
void dijkstra(vector<vector<int>> &g, int start) {
   
   
    const int n = g.size();// 顶点的个数。
    vector<int> dis(n, INT_MAX/2);// 每个点到起始点的距离
    dis[start] = 0;// 起始点到自己的值是 0 。
    vector<bool> visited(n, false); // 标记哪些顶点被访问过
    for (int i = 0; i < n; i++) {
   
   
        int k = -1;// 下一个没被标记且离起始点最近的顶点。
        int min = INT_MAX; // min 是 k 到起始点的值。
        for (int j = 0; j < n; j++) {
   
   // 寻找 k。
            if (!visited[j] && dis[j] < min) {
   
   
                min = dis[j];
                k = j;
            }
        }
        visited[k] = true;// 标记
        for (int j = 0; j < n; j++) {
   
   // 核心代码。
            // 顶点 j 没有被标记,并且 k 有到 j 的路径,并且这个路径更近,就更新。
            if (!visited[j] && g[k][j] != 0 && dis[k] + g[k][j] < dis[j])
                dis[j] = dis[k] + g[k][j];
        }
    }
    // 打印数组dis的值,测试使用。
    for (int di: dis)
        cout << di << ",";
}

3,迪杰斯特拉算法的堆优化
我们看到上面代码中外面的循环是遍历顶点,里面的循环主要是查找离起始点最近的顶点 v ,然后更新和 v 邻接的顶点。

如果这个图是个稀疏图,边特别少的话,在一个个查找很明显效率不高,所以在这种情况下可以使用最小堆来优化下,每次与顶点 v 邻接的点计算完之后把它加入到堆中,下次循环的时候直接弹出堆顶元素即可,它就是离起始点最近的点。

Java 代码:

/**
 * 使用堆优化的算法
 *
 * @param g     图的邻接矩阵
 * @param start 起始点
 */
public void dijkstra(int[][] g, int start) {
   
   
    int n = g.length;// 顶点的个数。
    int[] dis = new int[n];// 每个点到起始点的距离
    Arrays.fill(dis, Integer.MAX_VALUE >> 1);
    dis[start] = 0;// 起始点到自己的值是 0 。
    boolean visited[] = new boolean[n];// 标记哪些顶点被访问过
    // 创建堆,根据到起始点的距离排序
    PriorityQueue<int[]> pq = new PriorityQueue<>(Comparator.comparingInt(a -> a[0]));
    pq.offer(new int[]{
   
   0, start});// 起始点到它自己的距离是 0 。
    for (int i = 0; i < n; i++) {
   
   
        if (pq.isEmpty()) break; // 如果堆为空,退出循环
        // 每次出队都是离起始点最近且没被标记过的顶点。
        int k = pq.poll()[1];
        visited[k] = true;// 标记
        for (int j = 0; j < n; j++) {
   
   // 核心代码。
            // 顶点 j 没有被标记,并且 k 有到 j 的路径,并且这个路径更近,就更新。
            if (!visited[j] && g[k][j] != 0 && dis[k] + g[k][j] < dis[j]) {
   
   
                // 如果顶点 j 经过 k 到起始点的距离更近,就更新顶点 j 到
                // 起始点的距离,并把它添加到堆中。
                dis[j] = dis[k] + g[k][j];
                pq.offer(new int[]{
   
   dis[j], j});
            }
        }
    }
    // 打印数组dis的值,测试使用。
    for (int di : dis)
        System.out.print(di + ",");
}

C++ 代码:

void dijkstra(vector<vector<int>> &g, int start) {
   
   
    int n = g.size(); // 顶点的个数
    vector<int> dis(n, INT_MAX / 2); // 每个点到起始点的距离
    dis[start] = 0; // 起始点到自己的值是 0
    vector<bool> visited(n, false); // 标记哪些顶点被访问过
    // 创建堆,根据到起始点的距离排序
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<>> pq;
    pq.emplace(0, start); // 起始点到它自己的距离是 0
    for (int i = 0; i < n; i++) {
   
   
        // 每次出队都是离起始点最近且没被标记过的顶点
        if (pq.empty()) break; // 如果堆为空,退出循环
        int k = pq.top().second;
        pq.pop();
        visited[k] = true; // 标记
        for (int j = 0; j < n; j++) {
   
    // 核心代码
            // 顶点 j 没有被标记,并且 k 有到 j 的路径,并且这个路径更近,就更新
            if (!visited[j] && g[k][j] != 0 && dis[k] + g[k][j] < dis[j]) {
   
   
                // 如果顶点 j 经过 k 到起始点的距离更近,就更新顶点 j 到起始点的距离,并把它添加到堆中
                dis[j] = dis[k] + g[k][j];
                pq.emplace(dis[j], j);
            }
        }
    }

    // 打印数组dis的值,测试使用
    for (int di: dis)
        cout << di << ",";
    cout << endl;
}

4,为什么迪杰斯特拉算法不能处理带有负权边的图

为什么通过上述的操作可以保证得到的 dis 值最小?因为这里的图是没有负权边的,值只能越加越大,我们不断选择最小值进行标记然后更新和它邻接的点,即贪心的思路,最终保证起始点到每个顶点的值都是最小的。

如果有负权边在使用 Dijkstra 算法就行不通了,如下图所示,其中有负权边。

image.png
image.png

最后的结果是起始点到顶点 2 的值是 3 ,但实际上如果选择 0->1->2 这条路径的值是 2 ,会更小,所以有负权边并不适合 Dijkstra 算法。如果图是有环的可不可以使用 Dijkstra 算法呢?实际上只要没有负权边无论有环无环都是可以使用 Dijkstra 算法的。

如果有负权边该怎么解决呢?我们可以使用贝尔曼-福特算法(Bellman–Ford)和最短路径快速算法(Shortest Path Faster Algorithm:简称:SPFA),这两种算法虽然可以解决带有负权边的图,但不能解决有负权回路的图,关于这两种算法,后面我们也都会介绍。

这个是求最短路径的迪杰斯特拉算法,另外我还写了50多种《经典图论算法》,每种都使用C++和Java两种语言实现,熟练掌握之后无论是参加蓝桥杯,信奥赛,还是其他比赛,或者是面试,都能轻松应对。

image.png
image.png
image.png

相关文章
|
4天前
|
机器学习/深度学习 算法 Java
算法设计(动态规划应用实验报告)实现基于贪婪技术思想的Prim算法、Dijkstra算法
这篇文章介绍了基于贪婪技术思想的Prim算法和Dijkstra算法,包括它们的伪代码描述、Java源代码实现、时间效率分析,并展示了算法的测试用例结果,使读者对贪婪技术及其应用有了更深入的理解。
算法设计(动态规划应用实验报告)实现基于贪婪技术思想的Prim算法、Dijkstra算法
|
4天前
|
算法
路径规划算法 - 求解最短路径 - Dijkstra(迪杰斯特拉)算法
路径规划算法 - 求解最短路径 - Dijkstra(迪杰斯特拉)算法
11 0
|
1月前
|
算法
基于Dijkstra算法的最优行驶路线搜索matlab仿真,以实际城市复杂路线为例进行测试
使用MATLAB2022a实现的Dijkstra算法在城市地图上搜索最优行驶路线的仿真。用户通过鼠标点击设定起点和终点,算法规划路径并显示长度。测试显示,尽管在某些复杂情况下计算路径可能与实际有偏差,但多数场景下Dijkstra算法能找到接近最短路径。核心代码包括图的显示、用户交互及Dijkstra算法实现。算法基于图论,不断更新未访问节点的最短路径。测试结果证明其在简单路线及多数复杂城市路况下表现良好,但在交通拥堵等特殊情况下需结合其他数据提升准确性。
|
1月前
|
资源调度 算法 定位技术
|
2月前
|
存储 算法 数据可视化
Dijkstra算法在《庆余年》中的应用:范闲的皇宫之旅
Dijkstra算法在《庆余年》中的应用:范闲的皇宫之旅
|
2月前
|
存储 算法 C语言
数据结构学习记录——图-最短路径问题(无权图单源最短路径算法、有权图单源最短路径算法、多源最短路径算法、Dijkstra(迪杰斯特拉)算法、Floyd算法)
数据结构学习记录——图-最短路径问题(无权图单源最短路径算法、有权图单源最短路径算法、多源最短路径算法、Dijkstra(迪杰斯特拉)算法、Floyd算法)
49 1
|
2月前
|
算法 定位技术
探寻最短路径之谜:Dijkstra算法详解
探寻最短路径之谜:Dijkstra算法详解
|
5天前
|
算法
基于模糊控制算法的倒立摆控制系统matlab仿真
本项目构建了一个基于模糊控制算法的倒立摆控制系统,利用MATLAB 2022a实现了从不稳定到稳定状态的转变,并输出了相应的动画和收敛过程。模糊控制器通过对小车位置与摆的角度误差及其变化量进行模糊化处理,依据预设的模糊规则库进行模糊推理并最终去模糊化为精确的控制量,成功地使倒立摆维持在直立位置。该方法无需精确数学模型,适用于处理系统的非线性和不确定性。
基于模糊控制算法的倒立摆控制系统matlab仿真
|
5天前
|
机器学习/深度学习 算法 定位技术
MATLAB - 遗传算法(GA)求解旅行商问题(TSP)
MATLAB - 遗传算法(GA)求解旅行商问题(TSP)
9 3
|
6天前
|
算法
基于多路径路由的全局感知网络流量分配优化算法matlab仿真
本文提出一种全局感知网络流量分配优化算法,针对现代网络中多路径路由的需求,旨在均衡分配流量、减轻拥塞并提升吞吐量。算法基于网络模型G(N, M),包含N节点与M连接,并考虑K种不同优先级的流量。通过迭代调整每种流量在各路径上的分配比例,依据带宽利用率um=Σ(xm,k * dk) / cm来优化网络性能,确保高优先级流量的有效传输同时最大化利用网络资源。算法设定收敛条件以避免陷入局部最优解。