LeetCode 1014. 在 D 天内送达包裹的能力（C++、python）

传送带上的包裹必须在 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 <= D <= weights.length <= 50000

1 <= weights[i] <= 500

C++

class Solution {
public:
    int shipWithinDays(vector<int>& weights, int D) 
    {
        int n=weights.size();
        int left=*max_element(weights.begin(),weights.end());
        int right=accumulate(weights.begin(),weights.end(),0);
        int res=left;
        while(left<=right)
        {
            int mid=(left+right)/2;
            int count=0;
            int sum=0;
            for(int i=0;i<n;i++)
            {
                sum+=weights[i];
                if(sum>mid)
                {
                    count++;
                    sum=weights[i];
                }
            }
            count++;
            if(count<=D)
            {
                res=mid;
                right=mid-1;
            }
            else
            {
                left=mid+1;
            }
        }
        return res;       
    }
};

python

class Solution:
    def shipWithinDays(self, weights: List[int], D: int) -> int:
        n=len(weights)
        left=max(weights)
        right=sum(weights)
        res=left
        while left<=right:
            mid=(left+right)//2
            count=0
            su=0
            for i in range(n):
                su+=weights[i]
                if su>mid:
                    count+=1
                    su=weights[i]
            count+=1
            if count<=D:
                res=mid
                right=mid-1
            else:
                left=mid+1
        return res