【刷题打卡】1011. 在 D 天内送达包裹的能力

155 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

一、题目描述:

1011. 在 D 天内送达包裹的能力 - 力扣(LeetCode) (leetcode-cn.com)

传送带上的包裹必须在 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, 52 天:6, 73 天:84 天:95 天: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], days = 4
输出:3
解释:
第 1 天:1
第 2 天:2
第 3 天:3
第 4 天:1, 1

提示:

  • 1 <= days <= weights.length <= 5 * 10^4
  • 1 <= weights[i] <= 500

二、思路分析:

二分法

  1. 载重范围是[1, 50000*500],题目条件给出的限制
  2. 初始化l=1,r=50000*500
  3. 假设现在载重是m=(l+r)/2
    • 如果m可以满足条件,那么可以继续减少r来获取更小可能满足条件的m
    • 如果m不满足,说明太小,那么要加大l+1
    • 继续计算m

最后剩下一个问题如何判断”满足条件“,使用贪心法

  1. 用一个计数来记录需要运输的天数
  2. 遍历weights去计算和是否<=D, 大于等于则天数+1
  3. 判断天数<=D

三、AC 代码:


class Solution {
private:
    int n;
    int d;
    int maxWeight;

    bool Check(vector<int>& weights, int m)
    {
        int sum = 0;
        int cnt = 1;
        for (int i = 0; i < n; ++i)
        {
            if (weights[i] > m)
            {
                return false;
            }
            if (sum + weights[i] > m)
            {
                ++cnt;
                if (cnt > d)
                {
                    return false;
                }
                sum = 0;
            }
            sum += weights[i];
        }
        return true;
    }

public:
    int shipWithinDays(vector<int>& weights, int D) {
        n = weights.size();
        d = D;

        // 开始二分法
        int l = 1;
        int r = 5000*500;
        // 保存结果是满足条件的右边界
        int res = 0; 
        int m = 0;
        while (l < r)
        {
            m = (l+r) >> 1;
            if (Check(weights, m))
            {
                r = m;
            }
            else
            {
                l = m + 1;
            }
        }
        return r;
    }
};

四、总结:

这个题算是个中规中矩的二分法了。