分割数组的最大值
给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。
注意:
数组长度 n 满足以下条件:
1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)
解题思路
方法一动态规划
1、令 f[i][j]表示将数组的前 i个数分割为 j 段所能得到的最大连续子数组和的最小值。
2、进行状态转移时,我们可以考虑第 j段的具体范围,即我们可以枚举 k,其中前 k个数被分割为 j-1 段,而第 k+1到第 i 个数为第 j段。
3、这 j段子数组中和的最大值,就等于 f[k][j-1]与sub(k+1,i) 中的较大值,其中sub(i,j) 表示数组 nums 中下标落在区间 [i,j] 内的数的和。
4、状态转移方程
对于状态 f[i][j],由于我们不能分出空的子数组,因此合法的状态必须有 i≥j。对于不合法(i < ji<j)的状态,由于我们的目标是求出最小值,因此可以将这些状态全部初始化为一个很大的数。在上述的状态转移方程中,一旦我们尝试从不合法的状态 f[k][j−1] 进行转移,那么 max(⋯) 将会是一个很大的数,就不会对最外层的min{⋯} 产生任何影响。
5、f[0][0] 的值初始化为 0
6、最终的答案即为 f[n][m]
代码
class Solution:
def splitArray(self, nums: List[int], m: int) -> int:
#开数组
n = len(nums