二分查找之分割数组的最大值

LeetCode 410. 分割数组的最大值

给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。

注意:
数组长度 n 满足以下条件:
1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)
示例:

输入:
nums = [7,2,5,10,8]
m = 2

输出:
18

解释:
一共有四种方法将nums分割为2个子数组。
其中最好的方式是将其分为[7,2,5][10,8],
因为此时这两个子数组各自的和的最大值为18,在所有情况中最小。

思路:
首先分析题意,可以得出结论,结果必定落在**【max(nums), sum(nums)】**这个区间内,因为左端点对应每个单独的元素构成一个子数组,右端点对应所有元素构成一个子数组。

然后可以利用二分查找法逐步缩小区间范围,当区间长度为1时,即找到了最终答案。

每次二分查找就是先算一个mid值,这个mid就是代表当前猜测的答案,然后模拟一下划分子数组的过程,可以得到用这个mid值会一共得到的子区间数cnt,然后比较cnt和m的关系,来更新区间范围。

class Solution:
    def splitArray(self, nums: List[int], m: int) -> int:
        left = max(nums)
        right = sum(nums)
        while left <= right:
            mid = left + (right - left) // 2 # 最大和

            # 开始模拟切分数组
            temp = 0 # 初始第一个子数组和
            cnt = 1 # 数组个数初始化为1,说明还没有进行切分
            for num in nums:
                temp += num # 累加元素求和
                if temp > mid: #说明当前这个子数组的和已经超过了允许的最大值mid,需要把当前元素放在下一个子数组里
                    temp = num # 此时的num不能加入到当前子数组,要放到下一个子数组里
                    cnt += 1 # 切分了一个数组,cnt+1
            # 切分数组结束

            if cnt > m: # 说明分出了比要求多的子数组,多切了几刀,说明mid应该加大,这样能使子数组的个数减少
                left = mid + 1
            elif cnt < m:
                right = mid - 1
            elif cnt == m:
            	right = mid - 1
            
        return left

其他类似题目

LeetCode 875. 爱吃香蕉的珂珂

珂珂喜欢吃香蕉。这里有 N 堆香蕉,第 i 堆中有 piles[i] 根香蕉。警卫已经离开了,将在 H 小时后回来。

珂珂可以决定她吃香蕉的速度 K (单位:根/小时)。每个小时,她将会选择一堆香蕉,从中吃掉 K 根。如果这堆香蕉少于 K 根,她将吃掉这堆的所有香蕉,然后这一小时内不会再吃更多的香蕉。

珂珂喜欢慢慢吃,但仍然想在警卫回来前吃掉所有的香蕉。

返回她可以在 H 小时内吃掉所有香蕉的最小速度 K(K 为整数)。

示例 1:

输入: piles = [3,6,7,11], H = 8
输出: 4

示例 2:

输入: piles = [30,11,23,4,20], H = 5
输出: 30

示例 3:

输入: piles = [30,11,23,4,20], H = 6
输出: 23

思路:
最小速度是1,最大速度是max(piles),取中间速度,判断是否能在H小时内完成,如果可以,缩紧右边界,如果不可以,缩紧左边界,加快速度。

class Solution:
    def minEatingSpeed(self, piles: List[int], H: int) -> int:
       
        left = 1
        right = max(piles)
        while left <= right:
            mid = left + (right - left) // 2
            hour = 0
            for pile in piles:
                hour += pile // mid
                if pile % mid != 0:
                    hour += 1
            if hour > H: # 时间长于规定时间,所以需要加快速度,增大left
                left = mid + 1
            elif hour < H: # 时间少于规定时间,可以减慢速度,减少right
                right = mid - 1
            elif hour == H: # 时间等于规定时间,尝试减少right,因为要找最小速度
                right = mid - 1
        return left # 返回left,相当于找最大左边界

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

示例 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

思路

本题和第一个题分割数组最大值类似,基本上一样,利用二分法,最小值就是包裹的最大值,最大值是所有包裹的重量和,然后进行二分法求解,依次遍历,累加当前包裹重量,如果超过最大承载量,那么天数需要加1,最后判断天数是不是满足D天内,满足的话,收紧右边界,否则缩进左边界.
代码如下:

class Solution:
    def shipWithinDays(self, weights: List[int], D: int) -> int:
        left = max(weights)
        right = sum(weights)
        while left <= right:
            mid = left + (right - left) // 2
            days = 1
            temp = 0
            for weight in weights:
                temp += weight
                if temp > mid:
                    temp = weight
                    days += 1
            if days > D:
                left = mid + 1
            elif days < D:
                right = mid - 1
            elif days == D:
                right = mid - 1
        return left

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值