Leetcode 3381. Maximum Subarray Sum With Length Divisible by K

• Leetcode 3381. Maximum Subarray Sum With Length Divisible by K
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3381. Maximum Subarray Sum With Length Divisible by K

1. 解题思路

这一题的核心还是在于累积数组，然后按照K同余进行分组，我们考察累积数组当中对K同余的几个子序列，然后只需要考察这些子序列当中后值减去前值的最大值即可。

而对于后面这个问题，我们只需要顺序考察，不断更新前序数组当中的最小值，然后考察当前值减去前序最小值的差值是否更大即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def maxSubarraySum(self, nums: List[int], k: int) -> int:
        n = len(nums)
        s = list(accumulate(nums, initial=0))
        ans = -math.inf
        for i in range(k):
            _min = s[i]
            for j in range(i+k, n+1, k):
                ans = max(ans, s[j] - _min)
                _min = min(_min, s[j])
        return ans

提交代码评测得到：耗时262ms，占用内存44.8MB。