迪杰斯特拉算法

功能:求单源最短路径(有向图或者无向图)

思路:假设源点为k,维护k到所有节点的最短距离dist[]数组,以及已经找到最短路径的节点i的集合set。

1. 每次寻找不在set中的节点i,且dist[i]最小的节点i

2. 更新源点k通过i能到达的所有节点的最短路径dist

3. 将i存入set中

4. 循环上述操作直到set中包含所有节点

此处以LeetCode-743题为例,java实现代码如下:

public int networkDelayTime(int[][] times, int n, int k) {
        //graph[i]中的i表示起点,graph[i]存储i到某一节点的距离
        List<int[]>[] graph = new ArrayList[n+1];
        for (int i = 1; i < n+1; i++) {
            graph[i]=new ArrayList<>();
        }

        for (int i = 0; i < times.length; i++) {
            int begin=times[i][0];
            int end=times[i][1];
            int value=times[i][2];
            graph[begin].add(new int[]{end,value});
        }

        int[] dist = new int[n+1]; //k点到每个点的距离
        for (int i = 1; i < n+1; i++) {
            dist[i]=Integer.MAX_VALUE; //初始化所有距离为无穷大
        }
        dist[k]=0;//k到自己的距离为0
        HashSet<Integer> set = new HashSet<>();//存储已找到最短路径的节点
        int cur,minDist;
        while (set.size()<n){
            //寻找距离最小的点cur
            cur=-1;
            minDist=Integer.MAX_VALUE;
            for (int i = 1; i < n+1; i++) {
                if (minDist>dist[i]&&!set.contains(i)){
                    cur=i;
                    minDist=dist[i];
                }
            }
            if (cur==-1) return -1; //已经没有最小距离的点,此时存在点不可达,退出
            //将cur加入set
            set.add(cur);
            //更新源节点k通过cur走到其他节点的最短路径
            for (int i = 0; i < graph[cur].size(); i++) {
                int[] temp = graph[cur].get(i);
                int end=temp[0];
                int value=temp[1];
                dist[end]=Math.min(dist[end],dist[cur]+value);
            }
        }
        int res=0;
        for (int i = 1; i < n+1; i++) {
            res=Math.max(res,dist[i]);
        }
        return res==Integer.MAX_VALUE?-1:res;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值