迪杰斯特拉算法

迪杰斯特拉(Dijkstra)算法过程
设置出发顶点为v，顶点集合V{v1,v2,…}, v到V中各顶点的距离构成距离集合Dis,Dis{d1,d2,i…}, Dis集 合记录着v到图中各顶点的距离(到自身可以看作0, v到vi距离对应为di)

1） 从Dis中 选择值最小的di并移出Dis集合，同时移出V集合中对应的顶点vi,此时的v到vi即为最短路径
2） 更新Dis集合， 更新规则为:比较v到V集合中顶点的距离值，与v通过vi到V集 合中顶点的距离值，保留值较小的一个(同时也应该更新顶点的前驱节点为vi,表明是通过vi到达的)
3） 重复执行两步骤，直到最短路径项点为目标顶点即可结束

package com.dijkstra;

import java.util.Arrays;

/*
 * 迪杰斯特拉算法代码实现
*/
public class DijkstraAlgorithm {
    public static void main(String[] args) {
        char[] vertex = {'A','B','C','D','E','F','G'};
        //临接矩阵
        int[][] matrix = new int[vertex.length][vertex.length];
        final int N = 65535;  //表示不可以连接

        matrix[0] = new int[]{N, 5, 7, N, N, N, 2};
        matrix[1] = new int[]{5, N, N, 9, N, N, 3};
        matrix[2] = new int[]{7, N, N, N, 8, N, N};
        matrix[3] = new int[]{N, 9, N, N, N, 4, N};
        matrix[4] = new int[]{N, N, 8, N, N, 5, 4};
        matrix[5] = new int[]{N, N, N, 4, 5, N, 6};
        matrix[6] = new int[]{2, 3, N, N, 4, 6, N};
        //创建Graph对象
        Graph graph = new Graph(vertex, matrix);
        //测试图
        graph.show();

        graph.dsj(2);
        graph.showDijkstra();
    }
}

class Graph{
    private char[] vertex;
    private int[][] matrix;
    private VisitedVertex vv; //已经访问的顶点的集合

    public Graph(char[] vertex, int[][] matrix) {
        this.vertex = vertex;
        this.matrix = matrix;
    }

    //显示图
    public void show(){
        for (int i=0;i<vertex.length;i++){
            for (int j=0;j<vertex.length;j++){
                System.out.printf("%12d",matrix[i][j]);
            }
            System.out.println();
        }
    }

    //显示最终结果
    public void showDijkstra(){
        vv.show();
    }

    //迪杰斯特拉算法
    /**
     * @param index 对应出发顶点对应的下标
     * */
    public void dsj(int index){
        vv = new VisitedVertex(vertex.length, index);
        update(index);  //更新index顶点到周围顶点的距离和前驱顶点

        for (int j=1;j<vertex.length;j++){
            index = vv.updateArr();
            update(index);
        }
    }

    //更新index下标顶点到周围顶点的距离和周围顶点的前驱结点
    public void update(int index){
        int len = 0;
        //根据遍历我们邻接矩阵的matrix[index]行
        for (int i=0;i< matrix[index].length;i++){
            //len的含义：出发顶点到index顶点的距离 + index顶点到i顶点的距离的和
            len = vv.getDis(index) + matrix[index][i];
            if (!vv.in(i) && len < vv.getDis(i)){
                vv.updateDis(i,len);  //更新出发顶点到i顶点的距离
                vv.updatePre(i,index);  //更新i顶点的前驱为index顶点
            }
        }
    }
}

//已访问顶点集合
class VisitedVertex{
    //记录各个顶点是否访问过，访问过为1，没访问过为0，会动态更新
    public int[] already_arr;
    //每个下标对应的值为前一个顶点下标，会动态更新
    public int[] pre_visited;
    //记录出发顶点到其他顶点的距离，比如G为出发顶点，就会记录G到其他顶点的距离，会动态更新，求的最短距离就会存放到dis
    public int[] dis;

    /**
     * @param length   表示顶点的个数
     * @param index    出发顶点对应的下标，比如G顶点，对应下标为6
     * */
    public VisitedVertex(int length,int index) {
        this.already_arr = new int[length];
        this.pre_visited = new int[length];
        this.dis = new int[length];

        //初始化dis
        Arrays.fill(dis,65535);
        this.already_arr[index] = 1; //设置出发顶点被访问过
        this.dis[index] = 0; //设置出发顶点的访问距离为0
    }

    /**
     * 判断index顶点是否被访问过
     * @param index
     * @return 如果访问过，就返回true，否则返回false
     * */
    public boolean in(int index){
        return already_arr[index] == 1;
    }

    /**
     * 功能：更新出发顶点到index顶点的距离
     * @param index
     * @param len
     * */
    public void updateDis(int index,int len){
        dis[index] = len;
    }

    /**
     * 功能：更新pre这个顶点的前驱顶点为index结点
     * @param index
     * @param pre
     * */
    public void updatePre(int pre,int index){
        pre_visited[pre] = index;
    }

    /**
     * 返回index顶点到出发顶点的距离
     * */
    public int getDis(int index){
        return dis[index];
    }

    //继续选择并返回新的访问顶点，比如这里G完后，就是A作为新的访问顶点（注意不是出发顶点）
    public int updateArr(){
        int min = 65535,index = 0;
        for (int i=0;i<already_arr.length;i++){
            if (already_arr[i] == 0 && dis[i] < min){
                min = dis[i];
                index = i;
            }
        }
        //更新index顶点被访问过
        already_arr[index] = 1;
        return index;
    }
    //显示最后的结果
    //输出3个数组
    public void show(){
        System.out.println("==============================");
        for (int i : already_arr){
            System.out.print(i+" ");
        }
        System.out.println();
        for (int i :pre_visited){
            System.out.print(i+" ");
        }
        System.out.println();
        for (int i : dis){
            System.out.print(i+" ");
        }
        System.out.println();
        //为了更直观看出到每个点的距离，用代码美化一下
        int count = 0;
        char[] vertex = {'A','B','C','D','E','F','G'};
        for (int i : dis){
            if (i != 65535){
                System.out.print(vertex[count] + "("+i+")");
            }else {
                System.out.print("N");
            }
            count++;
        }
    }
}