迪杰斯特拉算法Java实现 Disjktra

关于迪杰斯特拉的理论知识，参考理解最短路径——迪杰斯特拉（dijkstra）算法

代码实现从源点到其他各点最短路径：

最重要的是每次循环找出离源点最近的未标记点 nearestPos，然后以此点为中间点，修正从 start 点经过 nearestPos 点到其他各点 i 的最短路径长度。

public class Dijkstra {
	static int MAX = 10000;

	public static void main(String[] args) {
		// 邻接矩阵
		int[][] weight = { { 0, 1, 5, MAX, MAX, MAX, MAX, MAX, MAX },
				{ 1, 0, 3, 7, 5, MAX, MAX, MAX, MAX },
				{ 5, 3, 0, MAX, 1, 7, MAX, MAX, MAX },
				{ MAX, 7, MAX, 0, 2, MAX, 3, MAX, MAX },
				{ MAX, 5, 1, 2, 0, 3, 6, 9, MAX },
				{ MAX, MAX, 7, MAX, 3, 0, MAX, 5, MAX },
				{ MAX, MAX, MAX, 3, 6, MAX, 0, 2, 7 },
				{ MAX, MAX, MAX, MAX, 9, 5, 2, 0, 4 },
				{ MAX, MAX, MAX, MAX, MAX, MAX, 7, 4, 0 } };
		int start = 0;
		int[] shortPath = dijsktra(weight, start);
		System.out.println("=====================================");
		for (int i = 0; i < shortPath.length; i++) {
			System.out.println("从" + start + "出发到" + i + "的最短距离为："	+ shortPath[i]);
		}
	}

	// 接受一个图的权重矩阵，和一个源点start（从0编号，顶点存在数组中）
	// 返回一个int[] 数组，表示从start到点i之间最短路径长度
	public static int[] dijsktra(int[][] matrix, int start) {
		int n = matrix.length;// 总共有n个顶点
		int[] shortPath = new int[n];// shortPath[i]存储从start到点i之间最短路径长度
		String[] path = new String[n];// path[i]表示从start到点i最短路径字符串表示
		boolean[] visited = new boolean[n];// visited[i]表示是否确定了从start到顶点i的最短路径

		shortPath[start] = 0;
		visited[start] = true;
		for (int i = 0; i < n; i++) {
			path[i] = start + "-->" + i;
		}

		// n个顶点，要进行n-1次循环。每次循环找出距离初始顶点start最近的未标记顶点，并以此顶点为中间点更新其它点最短路径长度
		for (int count = 1; count < n; count++) {
			// 选出一个距离初始顶点start最近的未标记顶点nearestPos
			int nearestPos = -1;
			int nearestLen = Integer.MAX_VALUE;
			for (int i = 0; i < n; i++) {
				if (!visited[i] && matrix[start][i] < nearestLen) {
					nearestLen = matrix[start][i];
					nearestPos = i;
				}
			}
			// 将新选出的顶点标记为已求出最短路径，且到start的最短路径就是nearestLen
			shortPath[nearestPos] = nearestLen;
			visited[nearestPos] = true;
			// 以nearestPos为中间点，修正从start到未访问各点的距离
			for (int i = 0; i < n; i++) {
				if (!visited[i] && matrix[start][nearestPos] + matrix[nearestPos][i] < matrix[start][i]) {
					matrix[start][i] = matrix[start][nearestPos] + matrix[nearestPos][i];
					path[i] = path[nearestPos] + "-->" + i;
				}
			}
		}
		
		for (int i = 0; i < n; i++) {
			System.out.println("从" + start + "出发到" + i + "的最短路径为：" + path[i]);
		}
		return shortPath;
	}

}

对应的图如下

程序输出结果

/*
从0出发到0的最短路径为：0-->0
从0出发到1的最短路径为：0-->1
从0出发到2的最短路径为：0-->1-->2
从0出发到3的最短路径为：0-->1-->2-->4-->3
从0出发到4的最短路径为：0-->1-->2-->4
从0出发到5的最短路径为：0-->1-->2-->4-->5
从0出发到6的最短路径为：0-->1-->2-->4-->3-->6
从0出发到7的最短路径为：0-->1-->2-->4-->3-->6-->7
从0出发到8的最短路径为：0-->1-->2-->4-->3-->6-->7-->8
=====================================
从0出发到0的最短距离为：0
从0出发到1的最短距离为：1
从0出发到2的最短距离为：4
从0出发到3的最短距离为：7
从0出发到4的最短距离为：5
从0出发到5的最短距离为：8
从0出发到6的最短距离为：10
从0出发到7的最短距离为：12
从0出发到8的最短距离为：16
*/

参考文章数据结构图之三（最短路径--迪杰斯特拉算法）