Leetcode学习笔记（八）在D天内送达快递的能力（二分法）

题目：传送带上的包裹必须在 D 天内从一个港口运送到另一个港口。传送带上的第 i 个包裹的重量为weights[i]。每一天，我们都会按给出重量的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。返回能在 D 天内将传送带上的所有包裹送达的船的最低运载能力。

示例：

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

思路：答案一定在区间[max{weights[i]},sum{weights[i]}]之间，所以直接二分答案检验可行性。

用二分查找的方式，来查找这个最低承载力，如果mid可以完成任务，我们把查找范围缩减至[lo,mid]（注意不是mid-1，因为mid可能是我们所求的解），否则我们去[mid+1,hi]区间中继续查找，详情见代码。

class Solution {
public:
    int shipWithinDays(vector<int>& weights, int D) {
        int sum_weight = 0;
        int max_weight = 0;
        int left, mid, right;
        for (int i = 0; i < weights.size(); i++) {
            if (weights[i] > max_weight) {
                max_weight = weights[i];
            }
            sum_weight += weights[i];
        }
        left = max_weight;   //二分法的两个临街值分别是数组的最大值和总和
        right = sum_weight;  

        while (left < right) {
            mid = left + ((right - left) >> 2); //可以用移位代替除法
            if (shipCanLoad(mid, weights, D))
                right = mid;    //注意不是mid-1,因为这时mid可能就是那个满足条件的临街值   
            else 
                left = mid + 1;
        }
        return left;
    }

    int shipCanLoad(int mid, vector<int>& weights, int D) {
        int tmp_weight = 0;
        int tmp_day = 0;
        for (int i = 0; i < weights.size(); i++) {
            tmp_weight += weights[i];
            if (tmp_weight > mid) {
                tmp_day++;    //如果所需重量累加起来大于mid，则需要下一天来装这次的货物
                tmp_weight = weights[i];
            }
        }
        return tmp_day < D;    //tmp_day < D满足所需天数就可以装下
    }
};