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) 对都 互不相同(即,不含重复边)

思路:

时间复杂度:O(n^2),空间复杂度O(n^2)

dijkstra算法的应用

代码:

func networkDelayTime(times [][]int, n int, k int) int {
    INF := 999999
    g := make([][]int, n+1)
    for i:=0;i<=n ;i++{
        g[i] = make([]int, n+1)
        for j:=0;j<=n;j++{
            g[i][j] = INF
        }
    }
    st := make([]bool,n+1)
    dist := make([]int, n+1)
    for _,time := range times{
        g[time[0]][time[1]] = time[2]
    }
    for i:=1;i<=n;i++{
        dist[i] = INF
    }
    dist[k] = 0
    for i:=0;i<n;i++{
        t := -1
        for j:=1; j<=n; j++{
            if !st[j] && (t == -1 || dist[j] < dist[t]){
                t = j
            }
        }

        st[t] = true
        for j:=1; j<=n; j++{
            dist[j] = min(dist[j], dist[t] + g[t][j])
        }
    }
    var res int
    for i:=1;i<=n;i++{
        res = max(res, dist[i])
    }
    if res == INF{
        return -1
    }
    
    return res
}

func min(a,b int)int{
    if a>b{
        return b
    }
    return a
}
func max(a,b int)int{
    if a>b{
        return a
    }
    return b
}

代码效率:

执行用时:44 ms, 在所有 Go 提交中击败了98.78%的用户
内存消耗:7.1 MB, 在所有 Go 提交中击败了91.46%的用户