用java写一个迪杰斯特拉算法
时间: 2023-03-20 07:01:30 浏览: 131
好的,以下是使用Java编写的Dijkstra算法示例代码:
```java
import java.util.*;
public class DijkstraAlgorithm {
public static void main(String[] args) {
int[][] graph = {{0, 7, 0, 5, 0, 0},
{7, 0, 8, 9, 7, 0},
{0, 8, 0, 0, 5, 0},
{5, 9, 0, 0, 15, 6},
{0, 7, 5, 15, 0, 8},
{0, 0, 0, 6, 8, 0}};
int start = 0;
int[] distances = dijkstra(graph, start);
System.out.println(Arrays.toString(distances));
}
public static int[] dijkstra(int[][] graph, int start) {
int n = graph.length;
int[] distances = new int[n];
Arrays.fill(distances, Integer.MAX_VALUE);
distances[start] = 0;
boolean[] visited = new boolean[n];
for (int i = 0; i < n - 1; i++) {
int minDistance = Integer.MAX_VALUE;
int minVertex = -1;
for (int j = 0; j < n; j++) {
if (!visited[j] && distances[j] < minDistance) {
minDistance = distances[j];
minVertex = j;
}
}
visited[minVertex] = true;
for (int j = 0; j < n; j++) {
if (!visited[j] && graph[minVertex][j] != 0 && distances[minVertex] + graph[minVertex][j] < distances[j]) {
distances[j] = distances[minVertex] + graph[minVertex][j];
}
}
}
return distances;
}
}
```
这个例子中,我们定义了一个图,然后使用Dijkstra算法计算从起始顶点到其他顶点的最短距离。算法的核心部分是两个for循环,第一个循环用于遍历所有的顶点,选择距离起始顶点最近的未访问过的顶点。第二个循环用于更新与该顶点相邻的所有未访问过的顶点的距离。最后返回distances数组,其中distances[i]表示从起始顶点到顶点i的最短距离。
阅读全文