C++数据结构与算法分析——Dijkstra算法

Dijkstra算法

介绍

Dijkstra算法是图论中用来求最短路的算法。这个最短路求的是起点1到终点n的最短距离dist[n]。它通过贪心的思想每次求出距离起点最短的点t，然后对其它所有未确定最短路的点的最短距离进行更新。当所有点都确定过最短路时，只需要将dist[n]取出即为起点1到终点n的最短路。

步骤

1. 初始化所有点到起点1的最短路dist[i] = ∞(0x3f3f3f3f)，dist[1] = 0(起点到起点距离为0)
2. 进行n次循环，每次求出一个点的最短路并更新其他路的最短路
3. 求出所有点中距离起点最近的点t，将t的状态更新为已确定最短路
4. 遍历$1\sim n$中所有点，更新点j到起点1的最短距离dist[j] = min(dist[j],dist[t] + g[t][j])，g[t][j]表示点t到点j的距离，即点1到点j的最短距离为点1到点j的距离dist[j]与点1到点t的最短路dist[t]和点t到点j的距离g[t][j]的和最小值。

时间复杂度$O(n^2)$

从上面的步骤中我们知道首先要进行n次循环，每次求出一个点的最短路，其次每次循环中我们都需要进行n次循环更新每个点的最短距离，因此时间复杂度为$O(n^2)$。

例题

题目描述

给定一个n个点m条边的有向图，图中可能存在重边和自环，所有边权均为正值。

请你求出1号点到n号点的最短距离，如果无法从1号点走到n号点，则输出 −1。

输入格式

第一行包含整数n和m。

接下来m行每行包含三个整数x,y,z，表示存在一条从点x到点y的有向边，边长为z。

输出格式

输出一个整数，表示1号点到n号点的最短距离。

如果路径不存在，则输出−1。

数据范围

$1\le n\le 500$,
$1\le m\le 105$,
图中涉及边长均不超过$10000$。

输入样例：

3 3
1 2 2
2 3 1
1 3 4

输出样例：

3

解题思路

这题是一个最短路问题，题目确保了所有距离均为正值，可以使用Dijkstra算法解答，需要解决重边和自环问题，首先自环问题在最后一步更新所有最短路时假如t == j，那么dist[j] >= dist[j] + g[t][j]恒成立，答案不会计入自环距离，重边问题只需要在最开始初始化g[t][j]时将最短的g[t][j]存下来即可。因为本题的图是稠密图，因此我们需要用邻接矩阵来存储图。

代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 510;

int g[N][N];//g[i][j]表示从i点到j点的距离
int dist[N];//dist[i]表示从1号点到i号点的最短距离
int n,m;
bool st[N];//标记第i点是否已算出最短路径

int dijkstra(){
    memset(dist,0x3f,sizeof dist);
    dist[1] = 0;//第1点距离初始化为0

    for(int i = 0; i < n; i++){
        int t = -1;
        for(int j = 1;j <= n; j++){
            //如果点j还没有算出最短路径 && t还是等于-1 或者 dist[t]不是最短
            //更新t = j
            if(!st[j] && (t == -1 || dist[t] > dist[j]))
                t = j;
        }

        st[t] = true;

        for(int j = 1; j <= n; j++)
            //点1到点j的距离 = min(点1到点j的距离, 点1到点t的距离 + 点t到点j的距离)
            dist[j] = min(dist[j],dist[t] + g[t][j]);
    }

    if(dist[n] == 0x3f3f3f3f) return -1;//如果没有路径，返回-1
    return dist[n];//如果有路径，返回距离
}

int main(){
    scanf("%d%d",&n,&m);

    memset(g,0x3f,sizeof g);//初始化每两点之间距离为正无穷

    while(m --){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);

        g[a][b] = min(g[a][b],c);//更新a - > b最短距离
    }

    printf("%d",dijkstra());

    return 0;
}