印象最深的一题应该是这个:给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。
主要思想是二分查找
题目要求的是分割成连续的子数组,并且结果为所有满足分组个数的最大和的最小值。仔细想想 最后的结果应该是介于原数组的最大值和原数组的和之间的某个值。
所以用二分查找。
low = max(nums) high = sum(nums) mid = (low+high) // 2
相当于向下取整。
假设此时mid
就是最后的结果,这时分割的子数组的个数num
与要求的m
进行比较。
- 如果
num > m
,说明此时的mid
比正确结果小了,因此应该在右区间继续查找; - 如果
num < m
,说明此时的mid
比正确结果大了,因此应该在左区间继续查找。
还有一种情况,如果分割的子数组的个数num == m
,此时也不一定是正确答案。正如示例给的那样,当num == m
时有4种分割方法。
原数组为 [7,2,5,10,8]
法1:[7] , [2,5,10,8]
结果为25
法2:[7,2] , [5,10,8]
结果为23
法3:[7,2,5]] , [10,8]
结果为18
法4:[7,2,5,10] , [8]
结果为24
因此,此时虽然num = m
,只要low < high
,就应该在右区间中继续找。
代码如下,不明白的可以手推一遍:
class Solution(object):
def splitArray(self, nums, m):
"""
:type nums: List[int]
:type m: int
:rtype: int
"""
left,right = max(nums),