最短路径之迪杰斯特拉算法的Java实现

java实现迪杰斯特拉算法—转载

package cn.yu.test;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		
		Scanner sc = new Scanner(System.in);
		ArrayList<ArrayList<Integer>> list = new ArrayList<>();
		String s = null;
		while(!(s=sc.nextLine()).equals("")) {
			String [] str = s.split(" ");
			ArrayList<Integer> arr = new ArrayList<Integer>();
			for(int i = 0;i<str.length;i++) {
				arr.add(Integer.parseInt(str[i]));
			}
			list.add(arr);
		}
		int len = list.size();
		int lie = list.get(0).size();
		int [][] weight= new int[len][lie];
		Iterator<ArrayList<Integer>> it = list.iterator();
		int i = 0;
		int j = 0;
		while(it.hasNext()) {
			ArrayList<Integer> li= it.next();
			for(int index:li) {
				weight[i][j++] = index; 
			}
			i++;
			j=0;
		 System.out.println();	
		}
		int [] shortPath = dijsktra(weight,0);
		for(int in:shortPath) {
			System.out.println(in);
		}
	}
	//接受一个有向图的权重矩阵，和一个起点编号start（从0编号，顶点存在数组中)
    //返回一个int[] 数组，表示从start到它的最短路径长度  
    public static int[] Dijsktra(int[][]weight,int start){
        int length = weight.length;
        int[] shortPath = new int[length];//存放从start到各个点的最短距离
        shortPath[0] = 0;//start到他本身的距离最短为0
        String path[] = new String[length];//存放从start点到各点的最短路径的字符串表示
        for(int i=0;i<length;i++){
            path[i] = start+"->"+i;
        }
        int visited[] = new int[length];//标记当前该顶点的最短路径是否已经求出，1表示已经求出
        visited[0] = 1;//start点的最短距离已经求出
        for(int count = 1;count<length;count++){
            int k=-1;
            int dmin = Integer.MAX_VALUE;
            for(int i=0;i<length;i++){
                if(visited[i]==0 && weight[start][i]<dmin){
                    dmin = weight[start][i];
                    k=i;
                }
            }
            //选出一个距离start最近的未标记的顶点     将新选出的顶点标记为以求出最短路径，且到start的最短路径为dmin。
            shortPath[k] = dmin;
            visited[k] = 1;
            //以k为中间点，修正从start到未访问各点的距离
            for(int i=0;i<length;i++){
                if(visited[i]==0 && weight[start][k]+weight[k][i]<weight[start][i]){
                    weight[start][i] = weight[start][k]+weight[k][i];
                    path[i] = path[k]+"->"+i;
                }
            }
        }
        for(int i=0;i<length;i++){
            System.out.println("从"+start+"出发到"+i+"的最短路径为："+path[i]+"="+shortPath[i]);
        }
        return shortPath;
         
    }
	
}