单源最短路之Dijkstra算法代码模板

常用算法代码模板总结

目录

注意事项

数组定义

实现步骤

练习例题：Dijkstra求最短路

---

注意事项

用来求单源最短路

所有边权都是正数，时间复杂度O(n^2)

可通过堆优化达到时间复杂度为O(mlogn)

数组定义

dist[i]数组：表示初始点到某个点的最短路径长度。

st[i]数组：表示初始点到i这个点的最短路径是否确定。

map[i][j]数组：存放点之间的关系，例如map[i][j]表示从i点到j点的距离，map[j][i]表示从j点到i点的距离，map[i][i]定义为0。

实现步骤

（1）首先遍历一遍查找到没有确定到初始点距离的点，并且这个点到初始点的距离应当是没有确定的点中到初始点距离最小的那个点。

（2）把步骤（1）找到的点标记为确定，然后用这个点来更新其他点到初始点的距离。

（有贪心的思想在里面，通过每次局部的最优，遍历完成后达到了整体最优）

练习例题：Dijkstra求最短路

 java题解

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	// 自环表示自己指向自己
	// 重边表示A到B有多条路可走，对于最短路而言，我们只需要保存距离最短的那条路
	static int[][] map = new int[510][510];// 存放点之间的关系
	static int[] dist = new int[510];// 存放点到初始点的距离
	static boolean[] st = new boolean[510];// 存放是否确定到初始点的距离最短
	static int n, m;// n表示节点数，m表示边数

	public static void Init() {// 初始化图中点与点的距离和保存的最短路径
		int len = map.length;
		Arrays.fill(dist, 10010);
		for (int i = 0; i < len; i++)
			Arrays.fill(map[i], 10010);
	}

	public static int Dijkstra() {

		dist[1] = 0;// 初始点到自己的距离为0

		for (int i = 0; i < n - 1; i++) {

			int temp = -1;
			for (int j = 1; j <= n; j++) {
				// 第一次替换找到没有标记确定的第一个点，之后循环找距离初始点最近的没有标记确定的点
				if (st[j] == false && (temp == -1 || dist[temp] > dist[j]))
					temp = j;
			}

			for (int j = 1; j <= n; j++) {
				// 通过上一步找到的点来更新这个点能够到达的其他点的距离
				dist[j] = Math.min(dist[j], dist[temp] + map[temp][j]);
				// map[temp][j]表示temp到j这个点的距离，再加上dist[temp]就表示初始点以temp为中间点到j的距离
			}
			// 更新完成之后，将该点标记为确定
			st[temp] = true;
		}
		if (dist[n] == 10010)
			return -1;
		return dist[n];

	}

	public static void main(String args[]) {
		Scanner input = new Scanner(System.in);
		n = input.nextInt();
		m = input.nextInt();
		Init();// 初始化
		for (int i = 0; i < m; i++) {
			int a = input.nextInt();// 从a到b距离c
			int b = input.nextInt();
			int c = input.nextInt();
			map[a][b] = Math.min(map[a][b], c);// 存在重边，取最小距离
		}

		System.out.println(Dijkstra());

	}
}