关于二分法在 leetcode 中的使用

引言

很多时候，算法题中都会出现最小化最大值或最大化最小值的要求。一般来说，这类题都可以使用二分法来解决。接下来，我们将结合 leetcode 中的题目对这一方法进行介绍。

leetcode 第1011题: 在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) 是不允许的。

对于这道题目，我们假设最终符合条件的最低运载能力为 x，那么所有大于 x 的运载能力都能完成任务，而所有小于 x 的运载能力都不能完成任务，只有这样，才能使运载能力最小化。因此我们可以先确定运载能力的范围，即在 min(weights) 和 sum(weights) 之间，然后我们就可以使用二分法的特性来确定范围。当然，最关键的是写一个函数来判断当前的运载能力能否完成任务。

代码

class Solution(object):
    def shipWithinDays(self, weights, D):
        """
        :type weights: List[int]
        :type D: int
        :rtype: int
        """
        def check(weights, D, load):
            need = 1
            cur = 0
            for weight in weights:
                if cur+weight > load:
                    need += 1
                    cur = 0
                cur += weight
            return need<=D
            
        
        start, end = max(weights), sum(weights)
        while(start<end):
            mid = (start+end)//2
            if check(weights, D, mid):
                end = mid
            else:
                start = mid+1
        return start

leetcode 第875题： 爱吃香蕉的珂珂

题目描述

珂珂喜欢吃香蕉。这里有 N 堆香蕉，第 i 堆中有 piles[i] 根香蕉。警卫已经离开了，将在 H 小时后回来。珂珂可以决定她吃香蕉的速度 K （单位：根/小时）。每个小时，她将会选择一堆香蕉，从中吃掉 K 根。如果这堆香蕉少于 K 根，她将吃掉这堆的所有香蕉，然后这一小时内不会再吃更多的香蕉。 珂珂喜欢慢慢吃，但仍然想在警卫回来前吃掉所有的香蕉。返回她可以在 H 小时内吃掉所有香蕉的最小速度 K（K 为整数）。
示例 1：
输入: piles = [3,6,7,11], H = 8
输出: 4
提示：
1 <= piles.length <= 10^4
piles.length <= H <= 10^9
1 <= piles[i] <= 10^9

这道题也是同样的道理，假设最小速度 K 能在 H 小时内吃掉所有香蕉，那么只要是大于这个速度的都能吃掉所有香蕉，反之小于这个速度的则不能。首先确定二分查找的范围在 1 和 max(piles) 之间，然后则是写出判断当前速度能否吃完所有香蕉的函数。

代码

class Solution:
    def minEatingSpeed(self, piles: List[int], h: int) -> int:

        def check(piles: List[int], h: int, k: int) -> bool:
            need = 0
            for pile in piles:
                need += (pile-1)//k + 1
            return need<=h 
        
        start, end = 1, max(piles)
        while(start<end):
            mid = (start+end)//2
            if check(piles, h, mid):
                end = mid
            else:
                start = mid + 1
        return start