【华为OD题库-064】最小传输时延I-java

题目

输出描述:
起点到终点得最小时延，不可达则返回-1
示例1:
输入:
3 3
1 2 11
2 3 13
1 3 50
1 3
输出:
24

思路

Dijkstra 算法，该算法B站视频讲解得较清楚
同leetcode： 743. 网络延迟时间

题解

package hwod;

import java.util.Arrays;
import java.util.Scanner;

public class TheLeastDelayTime {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(), m = sc.nextInt();
        int[][] nums = new int[m][3];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < 3; j++) {
                nums[i][j] = sc.nextInt();
            }
        }
        int start = sc.nextInt(), end = sc.nextInt();
        System.out.println(theLeastDelayTime(nums, n, start, end));
    }

    private static int theLeastDelayTime(int[][] nums, int n, int start, int end) {
        int[][] g = new int[n][n];
        final int INF = Integer.MAX_VALUE / 2;//防止越界
        for (int i = 0; i < n; i++) {
            Arrays.fill(g[i], INF);
        }
        for (int[] t : nums) {
            int x = t[0] - 1, y = t[1] - 1;
            g[x][y] = t[2];
        }
        int[] used = new int[n];
        int[] dist = new int[n];
        Arrays.fill(dist, INF);
        dist[start - 1] = 0;
        for (int i = 0; i < n; i++) {
            int x = -1;//未标记的距离start最近的节点
            //更新未标记的，距离start最近的节点
            for (int y = 0; y < n; y++) {
                if (used[y] == 0 && (x == -1 || dist[y] < dist[x])) {
                    x = y;
                }
            }
            used[x] = 1;
            for (int y = 0; y < n; y++) {
                dist[y] = Math.min(dist[y], dist[x] + g[x][y]);
            }
        }
        return dist[end - 1] == INF ? -1 : dist[end - 1];

    }
}

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。