迪杰斯特拉算法Dijkstra 求最短路径（java实现）

原本的有向图

代码

import java.util.Arrays;

public class Main {
    public static void main(String[] args) {
        int[][] graph = new int[][]{
                {0, 0, 5, 30, 0, 0},
                {2, 0, 0, 0, 8, 0},
                {0, 15, 0, 0, 0, 7},
                {0, 0, 0, 0, 0, 0},
                {0, 0, 0, 4, 0, 0},
                {0, 0, 0, 10, 18, 0}
        };

        DijkstraAlgorithm dijkstraAlgorithm = new DijkstraAlgorithm();
        dijkstraAlgorithm.dijkstra(graph, 0);
    }
}

class DijkstraAlgorithm {
    static final int V = 6; // 图中顶点的数量
    static final int INF = Integer.MAX_VALUE; // 表示无穷大

    // 找到未包括在最短路径树中的最小距离顶点的索引
    int minDistance(int[] dist, boolean[] sptSet) {
        int min = INF, minIndex = -1;

        for (int v = 0; v < V; v++) {
            if (!sptSet[v] && dist[v] <= min) {
                min = dist[v];
                minIndex = v;
            }
        }

        return minIndex;
    }

    // 打印最终的最短路径
    void printSolution(int[] dist) {
        System.out.println("源点A到各顶点的最短距离：");
        for (int i = 0; i < V; i++) {
            System.out.print((char) (i + 'A'));
            System.out.println(" \t\t " + dist[i]);
        }
    }

    // 执行迪杰斯特拉算法以找到从源顶点到其他顶点的最短路径
    void dijkstra(int[][] graph, int src) {
        int[] dist = new int[V]; // 存储最短路径的数组
        boolean[] sptSet = new boolean[V]; // 表示顶点是否包括在最短路径树中

        // 初始化所有距离为无穷大，且所有顶点都不在最短路径树中
        Arrays.fill(dist, INF);
        Arrays.fill(sptSet, false);

        // 源点到自身的距离为0
        dist[src] = 0;

        // 找到最短路径
        for (int count = 0; count < V - 1; count++) {
            int u = minDistance(dist, sptSet);

            sptSet[u] = true;

            //更新未被选中的节点的距离
            for (int v = 0; v < V; v++) {
                if (!sptSet[v] && graph[u][v] != 0 &&
                        dist[u] != INF &&
                        dist[u] + graph[u][v] < dist[v]) {
                    dist[v] = dist[u] + graph[u][v];
                }
            }
        }

        // 打印最短路径
        printSolution(dist);
    }
}

验证输出

红色的数字代表路径加入的顺序。

（by 归忆）