1011. 在 D 天内送达包裹的能力

Difficulty
Medium
Tags
二分查找
URL
https://leetcode-cn.com/problems/capacity-to-ship-packages-within-d-days/
Star
传送带上的包裹必须在 days 天内从一个港口运送到另一个港口。
传送带上的第 i 个包裹的重量为 weights[i]。每一天，我们都会按给出重量（weights）的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。
返回能在 days 天内将传送带上的所有包裹送达的船的最低运载能力。
示例 1：
输入：weights = [1,2,3,4,5,6,7,8,9,10], days = 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], days = 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 <= days <= weights.length <= 5 * 104
  • 1 <= weights[i] <= 500

法 1 二分法

思路
采用二分法猜测答案。思路同
🕹️
410. 分割数组的最大值
题解
class Solution:
    def shipWithinDays(self, weights: List[int], days: int) -> int:
        min_ = max(weights)
        max_ = sum(weights)
        
        # 找运载能力的左边界
        while min_ <= max_:
            mid = min_ + (max_ - min_) // 2
            # 如果 最低运载能力为 mid 不可行，要试探性的减小运载能力
            if self.isValid(weights, days, mid):
                max_ = mid - 1
            else:
                min_ = mid + 1

        return min_

    def isValid(self, weights, days, mid):
        # 最大运载能力为 mid ，判断是否可行
        count = 1
        cur_num = 0
        for w in weights:
            cur_num += w
            # 当前重量超过了 最低运载能力， 只能下一天再运载了
            if cur_num > mid:
                count += 1
                if count > days:
                    return False
                cur_num = w
        
        return True