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，在所有情况中最小。

解题思路

看了题解才做出来的，dp

用dp[i][j]表示前i个数字被划分成j段的结果，则转移方程：
$$dp[i][j]=min[\underset{k=1}{\overset{i-1}{\max }}(dp[k][j-1],sub\_sum(k+1,j))]$$

dp[i][j]与将前k个数字划分成j - 1段(即dp[k][j - 1])，还有第k + 1到第i个数字的和有关(即sub_sum(k + 1, j)前缀和)。
dp[i][j]实际上是这俩选项中的较大值，因为是“子数组和的最大值”
前k个数字，具体是前几个数字，就要在这些最大值中求最小值

边界值：

1. dp[0][0] = 0，将0个数字划分为0段，最终应该是0
2. dp[i][0]，不存在，即将多个数字划分为0段
3. dp[0][j]，不存在，将0个数字划分为多段
4. dp[i][j], j > i，不存在，如将3个数字划分为4段

因为目标是求最小值，所以对于不存在的值，初始化为1个最大的数

代码

class Solution:
    def splitArray(self, nums: List[int], m: int) -> int:
        pre_sum = [0]
        for each_num in nums:
            pre_sum.append(pre_sum[-1] + each_num)

        dp = [[2 ** 32] * (m + 1) for _ in range(len(nums) + 1)]
        dp[0][0] = 0
        for i in range(1, len(nums) + 1):
            for j in range(1, 1 + min(m, i)):
                for k in range(i):
                    dp[i][j] = min(dp[i][j], max(dp[k][j - 1], pre_sum[i] - pre_sum[k]))
        return dp[len(nums)][m]