Leetcode第128场周赛 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

解题思路

采用二分搜索的策略。要搜索的序列定义如下：船的最低运送能力为最重的物品，最高运送能力为一次运送所有物品。故序列的左右边界为[最重物品，物品总重]。对这个序列进行二分搜索，若选定点的运送能力可满足小于等于D送达所有货物，则收缩右边界，否则收缩左边界。具体实现见代码。

代码

class Solution {
public:
    bool check(int m, const vector<int> &w, const int &D)
    {
        int n = w.size();
        int cur = 0;
        int tot = 0;
        for (int i = 0; i < n; ++ i)
        {
            if (cur+w[i] > m)
            {
                tot ++;
                cur = w[i];
            }
            else
                cur += w[i];
        }
        if (cur > 0) ++ tot;
        return tot <= D;
    }
    int shipWithinDays(vector<int>& w, int D) {
        int L = *max_element(w.begin(), w.end());
        int R = 0;
        for (auto x : w) R += x;
        int ret = R;
        while (L <= R)
        {
            int m = (L+R)/2;
            if (check(m, w, D))
            {
                ret = m;
                R = m-1;
            }
            else
                L = m+1;
        }
        return ret;
    }
};

主程序逻辑如上述思路所示，由于货物重量均为整数，故运送能力增量可设为1。check()函数中，若船即将超载，则算作经过一天，统计需要运送的天数，返回需要天数小于D的逻辑判定。