LeetCode 1011. Capacity To Ship Packages Within D Days

LeetCode 1011. Capacity To Ship Packages Within D Days (在 D 天内送达包裹的能力)

题目

链接

https://leetcode-cn.com/problems/capacity-to-ship-packages-within-d-days/

问题描述

传送带上的包裹必须在 days 天内从一个港口运送到另一个港口。

传送带上的第 i 个包裹的重量为 weights[i]。每一天，我们都会按给出重量（weights）的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。

返回能在 days 天内将传送带上的所有包裹送达的船的最低运载能力。

示例

输入：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) 是不允许的。

提示

1 <= days <= weights.length <= 5 * 104
1 <= weights[i] <= 500

思路

与875类似，我们需要求的都是最小值，也就是左边界。

考虑到复杂度问题，采用二分查找，shiptime用于计算需要的天数。

复杂度分析

时间复杂度 O(nlogn)
空间复杂度 O(n)

代码

Java

    static int shiptime(int[] weights, int h) {
        int day = 1;
        int tmp = h;
        for (int i = 0; i < weights.length; i++) {
            if (tmp < weights[i]) {
                tmp = h;
                day++;
            }
            tmp -= weights[i];
        }
        return day;
    }

    public static int shipWithinDays(int[] weights, int days) {
        int left = 0;
        int right = 1;

        for (int weight : weights) {
            if (weight > left) {
                left = weight;
            }
            right+=weight;
        }
        while (left < right) {
            int mid = left + (right - left) / 2;
            int need = shiptime(weights, mid);
            if (need <= days) {
                right = mid;
            } else if (need > days) {
                left = mid + 1;
            }
        }
        return left;
    }