leet-code/src/main/java/leetcode/editor/cn/CapacityToShipPackagesWithinDDays.java

//传送带上的包裹必须在 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), (1
//0) 是不允许的。 
// 
//
// 示例 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 
// 
// Related Topics 数组 二分查找 
// 👍 212 👎 0

package leetcode.editor.cn;

//1011:在 D 天内送达包裹的能力
public class CapacityToShipPackagesWithinDDays {
    public static void main(String[] args) {
        //测试代码
        Solution solution = new CapacityToShipPackagesWithinDDays().new Solution();
        //15
        System.out.println(solution.shipWithinDays(new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, 5));
        //6
        System.out.println(solution.shipWithinDays(new int[]{3, 2, 2, 4, 1, 4}, 3));
        //3
        System.out.println(solution.shipWithinDays(new int[]{1, 2, 3, 1, 1}, 4));
    }

    //力扣代码
    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        public int shipWithinDays(int[] weights, int D) {
            int left = 0;
            int right = 0;
            for (int j : weights) {
                left = Math.max(left, j);
                right += j;
            }
            if (D == 1) {
                return right;
            }
            if (D == weights.length) {
                return left;
            }
            while (left < right) {
                int mid = (left + right) / 2;
                int days = 1;
                int sum = 0;
                for (int weight : weights) {
                    if (sum + weight > mid) {
                        days++;
                        sum = 0;
                    }
                    if (days > D) {
                        left = mid + 1;
                        break;
                    }
                    sum += weight;
                }
                if (days <= D) {
                    right = mid;
                }
            }
            return left;
//            // 官方
//            // 确定二分查找左右边界
//            int left = Arrays.stream(weights).max().getAsInt(), right = Arrays.stream(weights).sum();
//            while (left < right) {
//                int mid = (left + right) / 2;
//                // need 为需要运送的天数
//                // cur 为当前这一天已经运送的包裹重量之和
//                int need = 1, cur = 0;
//                for (int weight : weights) {
//                    if (cur + weight > mid) {
//                        ++need;
//                        cur = 0;
//                    }
//                    cur += weight;
//                }
//                if (need <= D) {
//                    right = mid;
//                } else {
//                    left = mid + 1;
//                }
//            }
//            return left;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}