- 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:continueseen.add(u)if t >= disappear[u]:continueif ans[u] != -1 and ans[u] <= t:continueans[u] = tfor v, l in graph[u]:if v in seen:continueif t+l >= disappear[v]:continueheapq.heappush(q, (t+l, v))return ans
提交代码评测得到:耗时1389ms,占用内存65.3MB。