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

本文涉及的产品
实时数仓Hologres,5000CU*H 100GB 3个月
智能开放搜索 OpenSearch行业算法版,1GB 20LCU 1个月
实时计算 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

相关文章
|
22天前
|
存储 人工智能 算法
数据结构与算法细节篇之最短路径问题:Dijkstra和Floyd算法详细描述,java语言实现。
这篇文章详细介绍了Dijkstra和Floyd算法,这两种算法分别用于解决单源和多源最短路径问题,并且提供了Java语言的实现代码。
58 3
数据结构与算法细节篇之最短路径问题:Dijkstra和Floyd算法详细描述,java语言实现。
|
1月前
|
存储 算法 程序员
迪杰斯特拉(Dijkstra)算法(C/C++)
迪杰斯特拉(Dijkstra)算法(C/C++)
|
3月前
|
机器学习/深度学习 算法 Java
算法设计(动态规划应用实验报告)实现基于贪婪技术思想的Prim算法、Dijkstra算法
这篇文章介绍了基于贪婪技术思想的Prim算法和Dijkstra算法,包括它们的伪代码描述、Java源代码实现、时间效率分析,并展示了算法的测试用例结果,使读者对贪婪技术及其应用有了更深入的理解。
算法设计(动态规划应用实验报告)实现基于贪婪技术思想的Prim算法、Dijkstra算法
|
3月前
|
算法
路径规划算法 - 求解最短路径 - Dijkstra(迪杰斯特拉)算法
路径规划算法 - 求解最短路径 - Dijkstra(迪杰斯特拉)算法
61 0
|
4月前
|
算法
基于Dijkstra算法的最优行驶路线搜索matlab仿真,以实际城市复杂路线为例进行测试
使用MATLAB2022a实现的Dijkstra算法在城市地图上搜索最优行驶路线的仿真。用户通过鼠标点击设定起点和终点,算法规划路径并显示长度。测试显示,尽管在某些复杂情况下计算路径可能与实际有偏差,但多数场景下Dijkstra算法能找到接近最短路径。核心代码包括图的显示、用户交互及Dijkstra算法实现。算法基于图论,不断更新未访问节点的最短路径。测试结果证明其在简单路线及多数复杂城市路况下表现良好,但在交通拥堵等特殊情况下需结合其他数据提升准确性。
|
4月前
|
资源调度 算法 定位技术
|
13天前
|
算法 安全 数据安全/隐私保护
基于game-based算法的动态频谱访问matlab仿真
本算法展示了在认知无线电网络中,通过游戏理论优化动态频谱访问,提高频谱利用率和物理层安全性。程序运行效果包括负载因子、传输功率、信噪比对用户效用和保密率的影响分析。软件版本:Matlab 2022a。完整代码包含详细中文注释和操作视频。
|
1月前
|
机器学习/深度学习 算法 数据安全/隐私保护
基于MSER和HOG特征提取的SVM交通标志检测和识别算法matlab仿真
### 算法简介 1. **算法运行效果图预览**:展示算法效果,完整程序运行后无水印。 2. **算法运行软件版本**:Matlab 2017b。 3. **部分核心程序**:完整版代码包含中文注释及操作步骤视频。 4. **算法理论概述**: - **MSER**:用于检测显著区域,提取图像中稳定区域,适用于光照变化下的交通标志检测。 - **HOG特征提取**:通过计算图像小区域的梯度直方图捕捉局部纹理信息,用于物体检测。 - **SVM**:寻找最大化间隔的超平面以分类样本。 整个算法流程图见下图。
|
10天前
|
人工智能 算法 数据安全/隐私保护
基于遗传优化的SVD水印嵌入提取算法matlab仿真
该算法基于遗传优化的SVD水印嵌入与提取技术,通过遗传算法优化水印嵌入参数,提高水印的鲁棒性和隐蔽性。在MATLAB2022a环境下测试,展示了优化前后的性能对比及不同干扰下的水印提取效果。核心程序实现了SVD分解、遗传算法流程及其参数优化,有效提升了水印技术的应用价值。
|
11天前
|
机器学习/深度学习 算法 数据安全/隐私保护
基于贝叶斯优化CNN-LSTM网络的数据分类识别算法matlab仿真
本项目展示了基于贝叶斯优化(BO)的CNN-LSTM网络在数据分类中的应用。通过MATLAB 2022a实现,优化前后效果对比明显。核心代码附带中文注释和操作视频,涵盖BO、CNN、LSTM理论,特别是BO优化CNN-LSTM网络的batchsize和学习率,显著提升模型性能。