给出整数数组 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
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/partition-array-for-maximum-sum 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
dp[i] 表示到 i 为止的最大和
class Solution {
public:
int maxSumAfterPartitioning(vector<int>& A, int K) {
int i, j, n = A.size(), Max;
int dp[n+1] = {0};
for(i = 1; i <= n; ++i)
{
Max = A[i-1];
for(j = 1; j <= K && i-j>=0; ++j)
{
Max = max(Max, A[i-j]);
dp[i] = max(dp[i], dp[i-j]+j*Max);
}
}
return dp[n];
}
};