[补][leetcode每日一题]4.19

1043. 分隔数组以得到最大和

提示

中等

255

相关企业

给你一个整数数组 arr，请你将该数组分隔为长度 最多 为 k 的一些（连续）子数组。分隔完成后，每个子数组的中的所有值都会变为该子数组中的最大值。

返回将数组分隔变换后能够得到的元素最大和。本题所用到的测试用例会确保答案是一个 32 位整数。

示例 1：

输入：arr = [1,15,7,9,2,5,10], k = 3
输出：84
解释：数组变为 [15,15,15,9,10,10,10]

示例 2：

输入：arr = [1,4,1,5,7,3,6,1,9,9,3], k = 4
输出：83

示例 3：

输入：arr = [1], k = 1
输出：1

提示：

• 1 <= arr.length <= 500
• 0 <= arr[i] <= 109
• 1 <= k <= arr.length

Solution

class Solution:
    def maxSumAfterPartitioning(self, arr: List[int], k: int) -> int:
        n = len(arr)
        f = [0] * (n + 1)
        for i in range(1, n + 1):
            mx = arr[i - 1]
            for j in range(k):
                if i - j - 1 >= 0:
                    tmp = arr[i - j - 1]
                    mx = max(tmp, mx)
                    f[i] = max(f[i], f[i - j - 1] + mx * (j + 1))
        # print(f)
        return f[-1]

dp，只需要考虑前面的k - 1个状态即可。所以完全没有必要弄二维数组