LeetCode 每日一题 743. 网络延迟时间

743. 网络延迟时间

有 n 个网络节点，标记为 1 到 n。

给你一个列表 times，表示信号经过 有向 边的传递时间。 times[i] = (ui, vi, wi)，其中 ui 是源节点，vi 是目标节点， wi 是一个信号从源节点传递到目标节点的时间。

现在，从某个节点 K 发出一个信号。需要多久才能使所有节点都收到信号？如果不能使所有节点收到信号，返回 -1 。

示例 1：

输入：times = [[2,1,1],[2,3,1],[3,4,1]], n = 4, k = 2
输出：2

示例 2：

输入：times = [[1,2,1]], n = 2, k = 1
输出：1

示例 3：

输入：times = [[1,2,1]], n = 2, k = 2
输出：-1

提示：

• 1 <= k <= n <= 100
• 1 <= times.length <= 6000
• times[i].length == 3
• 1 <= ui, vi <= n
• ui != vi
• 0 <= wi <= 100
• 所有 (ui, vi) 对都 互不相同（即，不含重复边）

方法一：Floyd（弗洛伊德）

思路

一个朴素的思路是使用「弗洛伊德」算法求得任意两点之间的最短路径，取其他点到 k 点的最大距离即可。

Floyd 算法三层循环搞定。

参考代码

public int networkDelayTime(int[][] times, int n, int k) {
    final int INF = Integer.MAX_VALUE / 2;

    // 初始化图 → 点到点的距离, 角标减1
    int[][] graph = new int[n][n];
    for (int i = 0; i < n; i++) {
        Arrays.fill(graph[i], INF);
        graph[i][i] = 0;
    }
    for (int[] t : times) {
        graph[t[0] - 1][t[1] - 1] = t[2];
    }

    for (int p = 0; p < n; p++) {
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                graph[i][j] = Math.min(graph[i][j], graph[i][p] + graph[p][j]);
            }
        }
    }

    int ret = Arrays.stream(graph[k - 1]).max().getAsInt();
    return ret == INF ? -1 : ret;
}

执行结果

方法二：Dijkstra（迪杰斯特拉）

思路

迪杰斯特拉算法详解 →_→ 视频

有三个变量需要了解

1. graph[][] 任一点到任一点的距离
2. dist[] 目标点 k 到其他点的距离
3. visit[] 已经处理过的点

参考代码

public int networkDelayTime(int[][] times, int n, int k) {
    // 定义无穷大
    final int INF = Integer.MAX_VALUE / 2;

    // 点到点之间的距离，下表从 0 开始
    int[][] graph = new int[n][n];
    for (int i = 0; i < n; i++) {
        Arrays.fill(graph[i], INF);
    }
    for (int[] t : times) {
        graph[t[0] - 1][t[1] - 1] = t[2];
    }
    
    // 其他点到 k 点的距离
    int[] dist = new int[n];
    Arrays.fill(dist, INF);
    dist[k - 1] = 0;
    
    // 访问过的点
    boolean[] visit = new boolean[n];
    for (int i = 0; i < n; i++) {
        // 寻找当前距离原点 k 最近的点
        int x = -1;
        for (int j = 0; j < n; j++) {
            if (!visit[j] && (x == -1 || dist[j] < dist[x])) {
                x = j;
            }
        }
        visit[x] = true;

        // 更新 x 点到其他点的距离
        for (int j = 0; j < n; j++) {
            dist[j] = Math.min(dist[j], dist[x] + graph[x][j]);
        }
    }

    int ret = Arrays.stream(dist).max().getAsInt();
    return ret == INF ? -1 : ret;
}

执行结果