刷题-力扣-1011. 在 D 天内送达包裹的能力

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

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/capacity-to-ship-packages-within-d-days/submissions/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

传送带上的包裹必须在 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

题目分析

1. 根据题目描述计算船最小运载能力，把传送带上的包裹按顺序在D天内运送完毕
2. 船的最小运载能力是要比最大的包裹的重量大，并且要大于包裹总重量除以D

代码

class Solution {
public:
    int shipWithinDays(vector<int>& weights, int D) {
        int sumWeight = 0;
        for (auto w : weights) sumWeight += w;
        int shipWeight = sumWeight % D == 0 ? sumWeight / D : sumWeight / D + 1;
        for (auto w : weights) shipWeight = shipWeight > w ? shipWeight : w;
        while (1) {
            int day = D;
            int weight = shipWeight;
            for (auto w : weights) {
                if (day <= 0) break;
                if (weight - w < 0) {
                    weight = shipWeight;
                    --day;
                }
                weight -= w;
            }
            if (day > 0) break;
            ++shipWeight;
        }
        return shipWeight;
    }
};