LeetCode-Python-1014. 在 D 天内送达包裹的能力

传送带上的包裹必须在 D 天内从一个港口运送到另一个港口。

传送带上的第 i 个包裹的重量为 weights[i]。每一天,我们都会按给出重量的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。

返回能在 D 天内将传送带上的所有包裹送达的船的最低运载能力。

示例 1:

输入:weights = [1,2,3,4,5,6,7,8,9,10], D = 5
输出:15
解释:
船舶最低载重 15 就能够在 5 天内送达所有包裹,如下所示:
第 1 天:1, 2, 3, 4, 5
第 2 天:6, 7
第 3 天:8
第 4 天:9
第 5 天:10

请注意,货物必须按照给定的顺序装运,因此使用载重能力为 14 的船舶并将包装分成 (2, 3, 4, 5), (1, 6, 7), (8), (9), (10) 是不允许的。 

示例 2:

输入:weights = [3,2,2,4,1,4], D = 3
输出:6
解释:
船舶最低载重 6 就能够在 3 天内送达所有包裹,如下所示:
第 1 天:3, 2
第 2 天:2, 4
第 3 天:1, 4

示例 3:

输入:weights = [1,2,3,1,1], D = 4
输出:3
解释:
第 1 天:1
第 2 天:2
第 3 天:3
第 4 天:1, 1

提示:

  1. 1 <= D <= weights.length <= 50000
  2. 1 <= weights[i] <= 500

思路:

二分查找,根据题意,结果一定落在【max(weights), sum(weights)】这个区间之间,因为左端点对应每天一条船,右端点对应只有一条超级大船。 然后利用二分法,每一次循环模拟运货的过程,然后根据需要的天数与 输入 D 的关系来调整区间左右端点。

扫描二维码关注公众号,回复: 5592119 查看本文章
class Solution(object):
    def shipWithinDays(self, weights, D):
        """
        :type weights: List[int]
        :type D: int
        :rtype: int
        """
        lo, hi = max(weights), sum(weights)
        while(lo <= hi):
            mid = (lo + hi) // 2 # mid 即为当前运送的capacity
            
            #------以下为模拟运货的过程,temp表示当前这条船承载的重量,day表示已用的天数------
            temp = 0
            day = 1
            for weight in weights:
                temp += weight
                if temp > mid:# 当前货运不动
                    day += 1
                    temp = weight
            #------以上为模拟运货的过程-----------------
            
            if day > D: # 当前的capacity太小了,不够,需要更大容量才能及时运完
                lo = mid + 1
            elif day <= D:
                hi = mid - 1

        return lo

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/88666260