Leetcode 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

 

这个就是经典的二分搜索问题,枚举这个极值即可,

这里一定要注意逻辑函数的细节,以及最后答案的调整输出。

class Solution {
public:
    bool canWeight(vector<int>& weights,int D, int load){
        int days=1;                        
        int round_weight = 0;
        for(int i=0;i<weights.size();i++){
            round_weight += weights[i];
            if(round_weight>load){
                days+=1;                          // 如果放入后不合法,放在后一天
                round_weight = weights[i];
                if(days>D)                        // 如果需要天数大于D天,返回false
                    return false;
            }
        }
        return true;
    }
    
    int shipWithinDays(vector<int>& weights, int D) {
        int max_load = 0;
        int max_weight = 0;
        for(int i=0;i<weights.size();i++){
            max_load+=weights[i];
            max_weight=max(max_weight,weights[i]);
        }
        int N = weights.size();
        int min_load = max(max_weight, max_load/N);
        if(D==1)
            return max_load;
        if(D>=weights.size())
            return min_load;
        while(min_load<=max_load){
            int mid = min_load + (max_load-min_load)/2;
            if(canWeight(weights,D,mid)){
                // 如果取一个mid能放下,说明这个值取很大
                max_load = mid-1;
            }
            else
                // 如果不能放下,说明这个值很小
                min_load = mid+1;
        }
        return min_load;
    }
};

 

点赞