Leetcode 3620. Network Recovery Pathways

1. 解题思路

这一题我最开始想的是遍历一下所有的网络路径,不过遇到了超时的情况。因此后来调整了一下处理思路,使用二分法的话就能够解决上述问题了。

二分法的思路的话就是给定任意一个值 k k k,考察如果只使用不小于 k k k的边,那么能否将 0 0 0节点与 n − 1 n-1 n1节点进行连接即可,然后我们二分查找出 k k k的上确界即可。

而对于每一个给定的 k k k的判断,我们使用一个深度优先遍历即可进行判断。

2. 代码实现

给出python代码实现如下:

class Solution:
    def findMaxPathScore(self, edges: List[List[int]], online: List[bool], k: int) -> int:
        n = len(online)
        if len(edges) == 0:
            return -1
        graph = defaultdict(list)
        for u, v, cost in edges:
            graph[u].append((v, cost))
        
        def is_possible(m):
            s = [(-0, 0)]
            while s:
                tot, u = heapq.heappop(s)
                tot = -tot
                if u == n-1:
                    return True
                for v, cost in graph[u]:
                    if online[v] == 0 or tot + cost > k or cost < m:
                        continue
                    heapq.heappush(s, (-tot-cost, v))
            return False
        
        i, j = min(cost for u, v, cost in edges), max(cost for u, v, cost in edges)
        if is_possible(j):
            return j
        elif not is_possible(i):
            return -1
        while j-i > 1:
            m = (i+j) // 2
            if is_possible(m):
                i = m
            else:
                j = m
        return i

提交代码评测得到:耗时890ms,占用内存65.74MB。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值