地铁最短路径dijkstra算法 java
时间: 2023-07-19 22:43:42 浏览: 127
好的,以下是地铁最短路径 Dijkstra 算法的 Java 代码实现:
```
import java.util.*;
public class Dijkstra {
private static final int INF = Integer.MAX_VALUE; // 无穷大
private static int[][] graph; // 地铁路线图
private static int[] dist; // 起点到各站点的距离
private static boolean[] visited; // 是否已经找到最短路径
/**
* dijkstra算法求起点到其他站点的最短距离
* @param start 起点
*/
public static void dijkstra(int start) {
int n = graph.length;
dist = new int[n];
visited = new boolean[n];
// 初始化距离数组
for (int i = 0; i < n; i++) {
dist[i] = INF;
}
dist[start] = 0;
// 循环n次,每次找出一个顶点的最短路径
for (int i = 0; i < n; i++) {
int minDist = INF;
int u = -1;
// 找出当前未确定最短路径的距离最小的点
for (int j = 0; j < n; j++) {
if (!visited[j] && dist[j] < minDist) {
minDist = dist[j];
u = j;
}
}
if (u == -1) {
break; // 所有点均已找到最短路径
}
visited[u] = true;
// 更新与u相邻的点的最短距离
for (int v = 0; v < n; v++) {
if (!visited[v] && graph[u][v] != INF && dist[u] + graph[u][v] < dist[v]) {
dist[v] = dist[u] + graph[u][v];
}
}
}
}
public static void main(String[] args) {
// 以下为测试数据,实际使用时需要根据实际地铁路线图进行修改
graph = new int[][]{
{0, 4, 2, INF, INF},
{4, 0, 1, 5, INF},
{2, 1, 0, 1, INF},
{INF, 5, 1, 0, 3},
{INF, INF, INF, 3, 0}
};
dijkstra(0);
System.out.println(Arrays.toString(dist)); // 输出起点到各站点的最短距离
}
}
```
以上代码中,我们通过一个邻接矩阵来表示地铁路线图,其中 INF 表示两个站点之间没有直接连接。在 dijkstra 方法中,我们先初始化起点到各站点的距离为无穷大,然后以此找出距离起点最近的未确定最短路径的点,将其标记为已经确定最短路径,同时更新与该点相邻的点的最短距离。重复该过程,直到所有点均已找到最短路径,最终得到起点到各站点的最短距离数组。
阅读全文