算法专题-迪杰斯特拉(Dijkstra)算法

力扣:743. 网络延迟时间

题目实际是求节点 KK 到其他所有点中最远的距离,那么首先需要求出节点 KK 到其他所有点的最短路,然后取最大值即可。

单源最短路问题可以使用 Dijkstra 算法,其核心思路是贪心算法。流程如下:

  1. 首先,Dijkstra 算法需要从当前全部未确定最短路的点中,找到距离源点最短的点 x。
  2. 其次,通过点 x 更新其他所有点距离源点的最短距离。例如目前点 A 距离源点最短,距离为 3;有一条 A->B 的有向边,权值为 1,那么从源点先去 A 点再去 B 点距离为 3 + 1 = 4,若原先从源点到 B 的有向边权值为 5,那么我们便可以更新 B 到源点的最短距离为 4。
  3. 当全部其他点都遍历完成后,一次循环结束,将 xx 标记为已经确定最短路。进入下一轮循环,直到全部点被标记为确定了最短路。

我们来看在实现时需要的代码支持:
首先,Dijkstra 算法需要存储各个边权,由于本题节点数量不超过 100,所以代码中使用了邻接矩阵 g[i][j] 存储从点 i 到点 j 的距离。若两点之间没有给出有向边,则初始化为 inf。算法还需要记录所有点到源点的最短距离,代码中使用了 dist[i] 数组存储源点到点 i 的最短距离,初始值也全部设为 inf。由于本题源点为 KK,所以该点距离设为 0。
其次,Dijkstra 算法需要标记某一节点是否已确定了最短路,在代码中使用了 used[i] 数组存储,若已确定最短距离,则值为 true,否则值为 false。
之所以 inf 设置为 INT_MAX / 2,是因为在更新最短距离的时候,要有两个距离相加,为了防止溢出 int 型,所以除以 2。

import java.util.Arrays;

public class Solution {
    public int networkDelayTime(int[][] times, int n, int k) {
        final int INF = Integer.MAX_VALUE / 2;
        //邻接矩阵存储边的信息
        int[][] g = new int[n][n];
        for (int i = 0; i < n; ++i) {
            Arrays.fill(g[i], INF);
        }
        for (int[] t : times) {
            //边序号从0开始
            int x = t[0] - 1, y = t[1] - 1;
            g[x][y] = t[2];
        }
        //从源点到某点的距离数组
        int[] dist = new int[n];
        Arrays.fill(dist, INF);
        //由于从k开始,所以该点距离设为0,也即源点
        dist[k - 1] = 0;
        //节点是否被更新数组
        boolean[] used = new boolean[n];

        for (int i = 0; i < n; ++i) {
            //在还未确定最短路的点中,寻找距离最小的点
            int x = -1;
            for (int y = 0; y < n; ++y) {
                if (!used[y] && (x == -1 || dist[y] < dist[x])) {
                    x = y;
                }
            }
            //用该点更新所有其他点的距离
            used[x] = true;
            for (int y = 0; y < n; ++y) {
                dist[y] = Math.min(dist[y], dist[x] + g[x][y]);
            }
        }
        //找到距离最短的点
        int ans = Arrays.stream(dist).max().getAsInt();
        return ans == INF ? -1 : ans;
    }

}

转载自https://leetcode-cn.com/problems/network-delay-time/solution/gtalgorithm-dan-yuan-zui-duan-lu-chi-tou-w3zc/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值