leetcode 1043. Partition Array for Maximum Sum

leetcode 1043. Partition Array for Maximum Sum

题意：分割一个数组，保证分割的每个段最多K个数，将每个段的每个数换成当前段最大的那个数，求最终的最大值。

思路：简单DP。dp[i]表示分割完前i个数，最大值是多少。

j表示当前段的长度[1,K]。

假设我们知道[i-j,i]这一段的最大值curMax，那么dp[i]可以从dp[i-j]推过来。

 dp[i] = max(dp[i],(i>=K?dp[i-j]:0)+curMax*j);

curMax又可以通过遍历j的时候来更新。

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