Leetcode-3381

Maximum Subarray Sum With Length Divisible by K

Time complexity: $O(n)$

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#include <iostream>
#include <vector>
#include <cmath>

class Solution {
public:
    long long maxSubarraySum(std::vector<int>& nums, int k) {
        int n = nums.size();
        std::vector<long long> dp(n);
        long long diff = 0;
        for (int i = 0; i < k; ++i) {
            diff += nums[i];
        }
        long long max_sum = dp[k - 1] = diff;
        
        for (int i = k; i < n; ++i) {
            diff += nums[i] - nums[i - k];
            dp[i] = std::max(dp[i - k], 0ll) + diff;
            max_sum = std::max(max_sum, dp[i]);
        }
        
        return max_sum;
    }
};

Leetcode-3381

Licensed under CC BY-NC-SA 4.0