1 问题描述
何为Dijkstra算法?
Dijkstra算法功能:给出加权连通图中一个顶点,称之为起点,找出起点到其它所有顶点之间的最短距离。
Dijkstra算法思想:采用贪心法思想,进行n-1次查找(PS:n为加权连通图的顶点总个数,除去起点,则剩下n-1个顶点),第一次进行查找,找出距离起点最近的一个顶点,标记为已遍历;下一次进行查找时,从未被遍历中的顶点寻找距离起点最近的一个顶点, 标记为已遍历;直到n-1次查找完毕,结束查找,返回最终结果。
2 解决方案
2.1 使用Dijkstra算法得到最短距离示例
此处借用文末参考资料1博客中一个插图(PS:个人感觉此图描述简单易懂):
2.2 具体编码
Dijkstra复杂度是O(N^2),如果用binary heap优化可以达到O((E+N)logN),用fibonacci heap可以优化到O(NlogN+E) 。
注意,Dijkstra算法只能应用于不含负权值的图。因为在大多数应用中这个条件都满足,所以这种局限性并没有影响Dijkstra算法的广泛应用。
其次,大家要注意把Dijkstra算法与寻找最小生成树的Prim算法区分开来。两者都是运行贪心法思想,但是Dijkstra算法是比较路径的长度,所以必须把起点到相应顶点之间的边的权重相加,而Prim算法则是直接比较相应边给定的权重。
下面的代码时间复杂度为O(N^2),代码中所用图为2.1使用Dijkstra算法得到最短距离示例中所给的图。
package com.liuzhen.chapter9;
public class Dijkstra {
/*
* 参数adjMatrix:为图的权重矩阵,权值为-1的两个顶点表示不能直接相连
* 函数功能:返回顶点0到其它所有顶点的最短距离,其中顶点0到顶点0的最短距离为0
*/
public int[] getShortestPaths(int[][] adjMatrix) {
int[] result = new int[adjMatrix.length]; //用于存放顶点0到其它顶点的最短距离
boolean[] used = new boolean[adjMatrix.length]; //用于判断顶点是否被遍历
used[0] = true; //表示顶点0已被遍历
for(int i = 1;i < adjMatrix.length;i++) {
result[i] = adjMatrix[0][i];
used[i] = false;
}
for(int i = 1;i < adjMatrix.length;i++) {
int min = Integer.MAX_VALUE; //用于暂时存放顶点0到i的最短距离,初始化为Integer型最大值
int k = 0;
for(int j = 1;j < adjMatrix.length;j++) { //找到顶点0到其它顶点中距离最小的一个顶点
if(!used[j] && result[j] != -1 && min > result[j]) {
min = result[j];
k = j;
}
}
used[k] = true; //将距离最小的顶点,记为已遍历
for(int j = 1;j < adjMatrix.length;j++) { //然后,将顶点0到其它顶点的距离与加入中间顶点k之后的距离进行比较,更新最短距离
if(!used[j]) { //当顶点j未被遍历时
//首先,顶点k到顶点j要能通行;这时,当顶点0到顶点j的距离大于顶点0到k再到j的距离或者顶点0无法直接到达顶点j时,更新顶点0到顶点j的最短距离
if(adjMatrix[k][j] != -1 && (result[j] > min + adjMatrix[k][j] || result[j] == -1))
result[j] = min + adjMatrix[k][j];
}
}
}
return result;
}
public static void main(String[] args) {
Dijkstra test = new Dijkstra();
int[][] adjMatrix = {
{0,6,3,-1,-1,-1},
{6,0,2,5,-1,-1},
{3,2,0,3,4,-1},
{-1,5,3,0,2,3},
{-1,-1,4,2,0,5},
{-1,-1,-1,3,5,0}};
int[] result = test.getShortestPaths(adjMatrix);
System.out.println("顶点0到图中所有顶点之间的最短距离为:");
for(int i = 0;i < result.length;i++)
System.out.print(result[i]+" ");
}
}
运行结果:
顶点0到图中所有顶点之间的最短距离为:
0 5 3 6 7 9
>
自己根据博客敲的
import java.util.Scanner;
public class Dijkstra1 {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()) {
int n = scanner.nextInt();
int[][] arr = new int[n][n];
int xian = scanner.nextInt();
for(int i = 0;i < xian;i++) {
int x = scanner.nextInt();
int y = scanner.nextInt();
int des = scanner.nextInt();
arr[x - 1][y - 1] = des;
arr[y - 1][x - 1] = des;
}
int L = 0;//当前走的路经长
int[] x = new int[n];//里面放走的城市
int bestL = Integer.MAX_VALUE;
int[] min = new int[1];//没办法值传递,int值改变不了,所以引用了数组
for(int i = 0;i < n;i++) {
x[i] = i;
}
min[0] = Integer.MAX_VALUE;
travel(arr,L,bestL,min,x,1);
System.out.println(min[0]);
}
}
public static void travel(int[][] arr,int L,int bestL,int[] min,int[] x,int index) {
if(index == arr.length) {
/*
//要回到初始点的话
if(arr[x[index - 1]][0] != 0 &&(L + arr[x[index - 1]][0] < bestL) ) {
bestL = L + arr[x[index - 1]][0];
if(bestL < min[0]) {
min[0] = bestL;
}
}
*/
//不回到初始点
if(L < min[0]) {
min[0] = L;
}
}else {
for(int i = index;i < arr.length;i++) {
if(arr[x[index - 1]][x[i]] != 0 && L + arr[x[index - 1]][x[i]] < bestL) {//就是说前一个城市可以到i这个城市还比当前到i还短就有机会让路径更短
swap(x,index,i);//就把当前城市与i调换让其先到i
L = L +arr[x[index - 1]][x[index]];
travel(arr,L,bestL,min,x,index + 1);
L = L - arr[x[index - 1]][x[index]];
swap(x,index,i);
}
}
}
}
public static void swap(int[] x,int i,int j) {
int temp = x[i];
x[i] = x[j];
x[j] = temp;
}
}