题目
给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环,所有边权均为正值
请你求出 1 号点到 n 号点的最短距离,如果无法从 1 号点走到 n 号点,则输出 −1
输入的第一行包含整数 n 和 m
接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z
输入:
3 3
1 2 2
2 3 1
1 3 4
输出:
3
public class 最短路_Dijkstra {
//初始化,g存储的是一个节点到另一个节点的距离,dist存储的是出发点到某一个节点的距离,st判断当前节点有没有被走过
//记住n和m在外面定义,要不然dijkstra调不到
public static int N = 510, n, m;
public static int[] dist = new int[N];
public static int[][] g = new int[N][N];
public static boolean[] st = new boolean[N];
public static void main(String[] args) {
//n是点数,m是边数,g最开始用正无穷覆盖
//a,b,c每次接受一个点到一个点的距离时,我们只保留最短的那条边即可,因为可能会有重边,太麻烦了
Scanner in = new Scanner(System.in);
n = in.nextInt(); m = in.nextInt();
for (int i = 1; i<=n; i++) {Arrays.fill(g[i], 0x3f3f3f3f);}
while (m-- > 0){
int a = in.nextInt(), b = in.nextInt(), c = in.nextInt();
g[a][b] = Math.min(g[a][b], c);
}
System.out.println(dijkstra());
}
public static int dijkstra() {
//dist初始也覆盖为正无穷,但节点1也就是起点,手动设置为0即可,自己到自己就是0嘛
Arrays.fill(dist, 0x3f3f3f3f);
dist[1] = 0;
//进行n次操作尝试抵达最终点
for (int i = 0; i<n; i++){
//每次用t拿到距离最近且未走过的节点下标
int t = -1;
for (int j = 1; j<=n; j++) {
if (!st[j] && (t == -1 || dist[t] > dist[j])) {t = j;}
}
st[t] = true; //用st标记t已经走过
//每次用t来对其他点更新距离
for (int j = 1; j<=n; j++){
dist[j] = Math.min(dist[j], dist[t] + g[t][j]);
}
}
//如果是正无穷代表没走到这个点,说明路走不通,返回-1,否则返回最终点到起点的距离
if (dist[n] == 0x3f3f3f3f) {return -1;}
else return dist[n];
}
}
如果不太理解,可以看这个视频:Dijkstra算法–最短路
这位up讲解的非常好,如果没接触过,纯看代码其实挺抽象的,视频还是很有助于理解的
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流