Python网络延迟时间（Dijkstra算法）

2024-01-04 05:00:15

即使是经验丰富的开发人员，在处理复杂算法时也可能会遇到障碍。本文将使用Python的Dijkstra算法，深入研究一个颇具挑战性的问题——网络延迟时间。我们将踏上一个引人入胜的旅程，深入了解算法的内部运作方式，并解决现实世界的难题。

Dijkstra算法概述

Dijkstra算法是一种贪婪算法，用于计算图中节点之间的最短路径。它从源节点开始，逐步探索图，直到到达目标节点。在每次迭代中，它选择具有最小权重（在这种情况下为延迟时间）的节点，并更新与该节点相邻节点的距离。

Python实现

import heapq

def network_delay_time(times, n, k):
    """
    :type times: List[List[int]]
    :type n: int
    :type k: int
    :rtype: int
    """
    graph = [[] for _ in range(n+1)]
    dist = [float('inf')] * (n+1)

    for u, v, w in times:
        graph[u].append((v, w))

    pq = [(0, k)]
    dist[k] = 0

    while pq:
        d, node = heapq.heappop(pq)
        if d > dist[node]:
            continue

        for neighbor, weight in graph[node]:
            new_dist = d + weight
            if new_dist < dist[neighbor]:
                dist[neighbor] = new_dist
                heapq.heappush(pq, (new_dist, neighbor))

    max_dist = max(dist[1:])
    return max_dist if max_dist != float('inf') else -1