Leetcode 3112. Minimum Time to Visit Disappearing Nodes

• Leetcode 3112. Minimum Time to Visit Disappearing Nodes
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3112. Minimum Time to Visit Disappearing Nodes

1. 解题思路

这一题的话思路上来说就是一个最优路径的问题，我们通过一个遍历，即可获得从0节点开始到其他所有连通节点的路径。

但是，单纯的遍历得到的路径会过多，而我们事实上只关心时间最优的一条路径即可，因此，我们使用堆排对遍历顺序进行一下选择，然后对遍历过程进行一下剪枝，即可得到我们最终的答案。

2. 代码实现

给出python代码实现如下：

class Solution:
    def minimumTime(self, n: int, edges: List[List[int]], disappear: List[int]) -> List[int]:
        graph = defaultdict(list)
        for u, v, l in edges:
            graph[u].append((v, l))
            graph[v].append((u, l))
            
        ans = [-1 for _ in range(n)]
        q = [(0, 0)]
        seen = set()
        
        while q:
            t, u = heapq.heappop(q)
            if u in seen:
                continue
            seen.add(u)
            if t >= disappear[u]:
                continue
            if ans[u] != -1 and ans[u] <= t:
                continue
            ans[u] = t
            for v, l in graph[u]:
                if v in seen:
                    continue
                if t+l >= disappear[v]:
                    continue
                heapq.heappush(q, (t+l, v))
        return ans

提交代码评测得到：耗时1389ms，占用内存65.3MB。