Leetcode 1043：分割数组以得到最大和

题目描述

给出整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    int maxSumAfterPartitioning(vector<int>& A, int K) {
        int len = A.size();
        vector<int> dp(len);
        for(int i=0;i<len;++i){
            int tmp = A[i];
            for(int j=1;j<=K&&i-j+1>=0;++j){
                tmp = max(tmp,A[i-j+1]);
                dp[i] = max(dp[i],((i-j>=0)?dp[i-j]:0)+j*tmp);
            }
        }
        return dp[len-1];
    }
};