1043. 分隔数组以得到最大和

给出整数数组 A,将该数组分隔为长度最多为 K 的几个(连续)子数组。分隔完成后,每个子数组的中的值都会变为该子数组中的最大值。

返回给定数组完成分隔后的最大和。

示例:

输入:A = [1,15,7,9,2,5,10], K = 3
输出:84
解释:A 变为 [15,15,15,9,10,10,10]

提示:

  1. 1 <= K <= A.length <= 500
  2. 0 <= A[i] <= 10^6

思路:动态规划,dp[i]表示前i+1个数的最大和,

dp[i]=max(dp[j]+ max(A[i-j : i]) *j ),其中A[i-j : i]这一段数组,是指A[i], A[i-1]……A[i-K+1]最多K个数,即A[i-j], j=[0, K-1]

class Solution {
public:
    int maxSumAfterPartitioning(vector<int>& A, int K) {
        vector<int>dp(A.size());
        for(int i=0; i<A.size(); ++i){
            int cur=0;
            for(int j=0; j<K && i-j>=0; ++j){
                cur=max(cur, A[i-j]);//不断地找到区间[A[i-j], A[i]]的最大值,区间长度为j+1,区间长度最大为K,j最大取K-1
                dp[i]=max(dp[i], (i-j-1>=0 ? dp[i-j-1] :0)+cur*(j+1));//前i+1个数的最大和=前i-j个数的最大和+区间最大值*区间长度
            }
        }
        return dp.back();
    }
};
发布了312 篇原创文章 · 获赞 32 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/Scarlett_Guan/article/details/104068367
今日推荐