Medium

Sum of K Subarrays With Length at Least MC++

Full explanation · Time O(k * n) · Space O(n)

// Time:  O(k * n)
// Space: O(n)

// prefix sum, dp
class Solution {
public:
    int maxSum(vector<int>& nums, int k, int m) {
        static const int NEG_INF = numeric_limits<int>::min();
        vector<int> prefix(size(nums) + 1);
        for (int i = 0; i < size(nums); ++i) {
            prefix[i + 1] = prefix[i] + nums[i];
        }
        vector<int> dp(size(nums) + 1, NEG_INF);
        dp[0] = 0;
        for (int i = 1; i <= k; ++i) {
            vector<int> new_dp(size(nums) + 1, NEG_INF);
            int mx = NEG_INF;
            for (int j = i * m - 1; j < size(nums); ++j) {
                mx = max(mx, dp[j - m + 1]);
                new_dp[j + 1] = prefix[j + 1] - prefix[j - m + 1] + mx;
                if (j != i * m - 1) {
                    new_dp[j + 1] = max(new_dp[j + 1], new_dp[j] + nums[j]);
                }
            }
            dp = move(new_dp);
        }
        return ranges::max(dp);
    }     
};