Bootstrap

Leetcode 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。

;