代码拉取完成,页面将自动刷新
# Time: O((|E| + |V|) * log|V|) = O(|E| * log|V|),
# if we can further to use Fibonacci heap, it would be O(|E| + |V| * log|V|)
# Space: O(|E| + |V|) = O(|E|)
import collections
import heapq
class Solution(object):
def reachableNodes(self, edges, M, N):
"""
:type edges: List[List[int]]
:type M: int
:type N: int
:rtype: int
"""
adj = [[] for _ in xrange(N)]
for u, v, w in edges:
adj[u].append((v, w))
adj[v].append((u, w))
min_heap = [(0, 0)]
best = collections.defaultdict(lambda: float("inf"))
best[0] = 0
count = collections.defaultdict(lambda: collections.defaultdict(int))
result = 0
while min_heap:
curr_total, u = heapq.heappop(min_heap) # O(|V|*log|V|) in total
if best[u] < curr_total:
continue
result += 1
for v, w in adj[u]:
count[u][v] = min(w, M-curr_total)
next_total = curr_total+w+1
if next_total <= M and next_total < best[v]:
best[v] = next_total
heapq.heappush(min_heap, (next_total, v)) # binary heap O(|E|*log|V|) in total
# Fibonacci heap O(|E|) in total
for u, v, w in edges:
result += min(w, count[u][v]+count[v][u])
return result
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。