提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
题目描述
给你一个整数数组 arr,请你将该数组分隔为长度最多为 k 的一些(连续)子数组。分隔完成后,每个子数组的中的所有值都会变为该子数组中的最大值。
返回将数组分隔变换后能够得到的元素最大和。
注意,原数组和分隔后的数组对应顺序应当一致,也就是说,你只能选择分隔数组的位置而不能调整数组中的顺序。
示例 1:
输入:arr = [1,15,7,9,2,5,10], k = 3
输出:84
解释:
因为 k=3 可以分隔成 [1,15,7] [9] [2,5,10],结果为 [15,15,15,9,10,10,10],和为 84,是该数组所有分隔变换后元素总和最大的。
若是分隔成 [1] [15,7,9] [2,5,10],结果就是 [1, 15, 15, 15, 10, 10, 10] 但这种分隔方式的元素总和(76)小于上一种。
示例 2:
输入:arr = [1,4,1,5,7,3,6,1,9,9,3], k = 4
输出:83
示例 3:
输入:arr = [1], k = 1
输出:1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-array-for-maximum-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题过程
解题思路
动态规划:dp[i]表示前i个数组元素能够达成目标的最大值。从后往前推,假设已知前i - 1个数组的dp结果,则第i个结果可能是dp[i-1]和arr[i]的和,或者dp[i-2]与2*Max{arr[i,i-1]}的和,依次类推。状态转移方程:dp[i] = Math.max(dp[i], dp[j] + (i - j) * max);
class Solution {
public int maxSumAfterPartitioning(int[] arr, int k) {
int dp [] = new int[arr.length + 1];
for(int i = 0; i <= arr.length; i++){
int j = i - 1;
int max = -1;
while(i - j <= k && j >= 0){
max = Math.max(arr[j], max);
dp[i] = Math.max(dp[i], dp[j] + (i - j) * max);
j--;
}
}
return dp[arr.length];
}
}