leetcode1043. 分隔数组以得到最大和/动态规划

题目：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）
链接：添加链接描述
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

基本思想：动态规划

• dp[i] : 前 i 个元素包括第 i 个元素 分隔后的最大和
• 状态：数组中的每一个元素
• 选择：考虑当前元素的情况下，其包含该元素的左面连续K个元素如何分隔
• 状态转移方程：针对当前元素左面的连续K个元素，考虑每一种可能的分隔情况的最大值。$$dp[i]=max(dp[i],max(A[j\dots i])\ast (i-j+1)+dp[j-1]),i-K<j<=i并且j>=0$$

说明：代码中内层循环从后往前遍历，便于找最大值

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