Java 实现堆优化的dijkstra算法


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

public class dijkstra_queue {
    static int INF = 0x3f3f3f3f;
    //将权值按升序排序
    static Comparator<Node>com = new Comparator<>() {
        public int compare(Node o1, Node o2) {
            return o1.val-o2.val;
        }
    };
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();//边数
        int end = sc.nextInt();//终点
        int[] dis = new int[n];//权值
        int[][] map = new int[n][n];//图
        
        //邻接矩阵存图
        for(int i = 0;i<n;i++){
            for(int j = 0;j<n;j++){
                map[i][j] = INF;
            }
        }
        for(int i = 0;i<m;i++){
            int n1 = sc.nextInt();
            int n2 = sc.nextInt();
            int v = sc.nextInt();
            //无向图
            map[n2][n1] = v;
            map[n1][n2] = v;
        }
       	//起点为0
        dijkstra(0,dis,map);
        int ans = dis[end];
        System.out.println(ans);
    }
    //核心代码
    public static void dijkstra(int start,int[] dis,int[][] map){
        boolean[] vis = new boolean[map.length];
        Queue<Node> pq = new PriorityQueue<>(com);//构造小顶堆
        pq.add(new Node(start,0));//将起点add到堆中
		//初始化
        for(int i = 0;i < map.length;i++){
            vis[i] = false;
            dis[i] = INF;
        }
        dis[start] = 0;

        while(!pq.isEmpty()){
            Node 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 Node(i,dis[i]));
                }
            }
        }
    }
}
//节点类
class Node {
    public int side;
    public int val;
    public Node(int side, int val) {
        this.side = side;
        this.val = val;
    }
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Uranus^

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值