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