dijkstra堆优化——Java（最短路）

最新推荐文章于 2024-08-10 23:24:32 发布

Uranus^

最新推荐文章于 2024-08-10 23:24:32 发布

阅读量484

点赞数 1

文章标签： java 算法蓝桥杯

本文链接：https://blog.csdn.net/intwei/article/details/123862912

版权

问题描述:
　给定n个结点两两之间的单向边的长度，求第一个点到其他点的最短路径。
输入格式:
　　输入第一行包含一个整数n，表示点数。
　　接下来n行，每行包含n个整数，第i行表示第i个点到每个点的边的长度，如果没有边，则用0表示。
输出格式:
　　输出1行，包含n个数，表示第1个点到其他点的最短路径长度，如果没有可达的路径，则输出-1。
样例输入:
3
0 1 0
0 0 6
0 2 0
样例输出:
0 1 7


import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;

public class Main {
    static int INF = 0x3f3f3f3f;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] dis = new int[n];
        int[][] map = new int[n][n];
        //邻接矩阵存图
        sc.nextLine();
        for (int i = 0; i < n; i++) {
            String[] split = sc.nextLine().split(" ");
            for (int j = 0; j < n; j++) {
                if (Integer.parseInt(split[j]) == 0) {
                    map[i][j] = INF;
                } else {
                    map[i][j] = Integer.parseInt(split[j]);
                }
            }
        }
        dijkstra(0,dis,map);
        for(int i = 0;i<n;i++){
            if(i == 0){
                System.out.print(0 + " ");
            }else if(dis[i] == INF){
                System.out.print("-1 ");
            }else {
                System.out.print(dis[i] + " ");
            }
        }
    }
    public static void dijkstra(int start,int[] dis,int[][] map){
        boolean[] vis = new boolean[map.length];
        Queue<Nodeing> pq = new PriorityQueue<>((n1,n2)->n1.val-n2.val);
        pq.add(new Nodeing(start,0));

        for(int i = 0;i < map.length;i++){
            vis[i] = false;
            dis[i] = INF;
        }
        dis[start] = 0;

        while(!pq.isEmpty()){
            Nodeing node = pq.poll();
            int bian = node.side;
            if(vis[bian] == true){
                continue;
            }
            vis[bian] = true;
            for(int i = 0;i<map.length;i++){
                if(vis[i] == false && map[bian][i] + node.val < dis[i]){
                     dis[i] = map[bian][i] + node.val;
                     pq.add(new Nodeing(i,dis[i]));
                }
            }
        }
    }
}
class Nodeing {
    public int side;
    public int val;
    public Nodeing(int side, int val) {
        this.side = side;
        this.val = val;
    }
}