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

题目链接

1043. 分隔数组以得到最大和 - 力扣（Leetcode）

给你一个整数数组 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

Java代码

class Solution {
    public int maxSumAfterPartitioning(int[] arr, int k) {
        //dp用来保存第i项时前i个数组按照规则得到的结果，则数组最后一项为arr数组的结果
        int[] dp = new int[arr.length + 1];
        for(int i = 1; i <= arr.length; i++){
            //记录一组数字的最大值
            int maxValue = arr[i - 1];
            //向前取几个数字，代表前几个数字为一组，则dp[j]+maxValue*（取几个）最大的那个填在dp[i]中
            for (int j = i -1; j >= 0 && j >= i - k; j--){
                //更新最大的值
                maxValue = Math.max(maxValue, arr[j]);
                dp[i] = Math.max(dp[i], dp[j] + maxValue * (i - j));
            }
        }
        return dp[arr.length];
    }
}