Java数据结构与算法笔记——带权图最短路径（Dijkstra算法）

算法思路

1. 指定一个节点，比如我们要计算A到其他节点的最短路径。

2. 引入两个集合（S、U），S集合包含已求出的最短路径的点（以及相应的最短长度），U集合包含未求出最短路径的点（以及A到该点的路径，如果没有直接相连，初始为∞）

3. 初始化两个集合，S初始化时只有A→A = 0，U集合初始化时为A→B = 50，A→C = ∞，A→D = 80，A→E = ∞。接下来要进行核心两个步骤了。

4. 从U集合中找出路径最短的点，加入到S集合，例如A→B = 50。

5. 更新U集合路径，if(B到D,C,E的距离 + AB距离 < A到D,C,E的距离)则更新U里的对应数据。

6. 循环执行4,5两步骤，直至遍历结束，得到A到其他节点的最短路径。

算法实现

package shortestpath;

public class Dijkstra {
    private static int M = 10000; //定义一个很大的数，表示此路不通

    /**
     * main方法
     */
    public static void main(String[] args) {
        int[][] weight1 = {//无向图邻接矩阵
                {0,3,M,7,M},
                {3,0,4,2,M},
                {M,4,0,5,4},
                {7,2,5,0,6},
                {M,M,4,6,0}
        };

        int[][] weight2 = {//有向图邻接矩阵
                {0,10,M,30,100},
                {M,0,50,M,M},
                {M,M,0,M,10},
                {M,M,20,0,60},
                {M,M,M,M,0}
        };

        int[][] weight3 = {//无向图邻接矩阵
                {0,3,2,M,M,M},
                {3,0,1,4,3,M},
                {2,1,0,5,M,M},
                {M,4,5,0,3,M},
                {M,3,M,3,0,1},
                {M,M,M,M,1,0}
        };

        int start=0;
        int[] shortPath = dijkstra(weight3,start);

        for(int i = 0;i < shortPath.length;i++)
            System.out.println("从"+start+"出发到"+i+"的最短距离为："+shortPath[i]);
    }

    /**
     * 实现Dijkstra算法的方法
     * @param weight 邻接矩阵
     * @param start 求start顶点到其他顶点的最短路径
     */
    public static int[] dijkstra(int[][] weight, int start) {
        //接受一个有向图的权重矩阵，和一个起点编号start（从0编号，顶点存在数组中）
        //返回一个int[] 数组，表示从start到它的最短路径长度
        int n = weight.length;//顶点个数
        int[] shortPath = new int[n];//保存start到其他各点的最短路径
        String[] path = new String[n];//保存start到其他各点最短路径的字符串表示
        for(int i=0;i<n;i++){
            path[i]=new String(start+"-->"+i);
        }
        int[] visited = new int[n];   //标记当前该顶点的最短路径是否已经求出,1表示已求出

        //初始化，第一个顶点已经求出
        shortPath[start] = 0;
        visited[start] = 1;

        for(int count=1; count<n; count++) {   //要加入n-1个顶点
            int k = -1;//选出一个距离初始顶点start最近的未标记顶点
            int dmin = Integer.MAX_VALUE;

            //找到一个距离start最近的节点以及这个最近的距离
            for(int i=0; i<n; i++) {
                if(visited[i] == 0 && weight[start][i] < dmin) {
                    //如果到这个点的最短距离这个点还未被求出，并且start节点到该点的距离小于目前已知的最小距离
                    dmin = weight[start][i];
                    k = i;
                }
            }

            //将新选出的顶点标记为已求出最短路径，且到start的最短路径就是dmin
            shortPath[k] = dmin;
            visited[k] = 1;

            //以k为中间点，修正从start到未访问各点的距离
            for(int i = 0; i < n; i++) {
                if(visited[i] == 0 && weight[start][k] + weight[k][i] < weight[start][i]) {
                    //如果还未算出到节点i的最短距离&&从start以k为中间节点到i的距离小于从start到i的距离
                    weight[start][i] = weight[start][k] + weight[k][i];
                    path[i] = path[k] + "-->" + i;
                }
            }
        }

        for(int i = 0; i < n; i++) {
            System.out.println("从"+start+"出发到"+i+"的最短路径为："+path[i]);
        }
        System.out.println("=====================================");
        return shortPath;
    }
}