题目描述:
给出整数数组 A,将该数组分隔为长度最多为 K 的几个(连续)子数组。分隔完成后,每个子数组的中的值都会变为该子数组中的最大值。
返回给定数组完成分隔后的最大和。
示例:
输入:A = [1,15,7,9,2,5,10], K = 3
输出:84
解释:A 变为 [15,15,15,9,10,10,10]
提示:
1 <= K <= A.length <= 500
0 <= A[i] <= 10^6
没写出来参照别人的
代码
class Solution {
public int maxSumAfterPartitioning(int[] A, int K) {
int[] result = new int[A.length];
for (int i = 0; i < A.length; i++) {
int maxValue = 0, maxSum = 0;
for (int j = i; j >= 0 && i - j + 1 <= K; j--) {
if (maxValue < A[j]) {
maxValue = A[j];
}
int tempSum = (j == 0 ? 0 : result[j - 1]) + (i - j + 1) * maxValue;
if (maxSum < tempSum) {
maxSum = tempSum;
}
}
result[i] = maxSum;
}
return result[result.length - 1];
}
}