Advertisement
3381. Maximum Subarray Sum With Length Divisible by K
MediumView on LeetCode
Time: O(n)
Space: O(k)
Approach
Prefix sums mod k tracking; for each prefix[i] add current prefix minus min prefix of same mod.
3381.cs
C#
// Approach: Prefix sums mod k tracking; for each prefix[i] add current prefix minus min prefix of same mod.
// Time: O(n) Space: O(k)
public class Solution
{
public long MaxSubarraySum(int[] nums, int k)
{
long ans = long.MinValue;
long prefix = 0;
long[] minPrefix = new long[k];
for (int i = 0; i < k; i++)
minPrefix[i] = long.MaxValue / 2;
minPrefix[k - 1] = 0;
for (int i = 0; i < nums.Length; ++i)
{
prefix += nums[i];
ans = Math.Max(ans, prefix - minPrefix[i % k]);
minPrefix[i % k] = Math.Min(minPrefix[i % k], prefix);
}
return ans;
}
}Advertisement
Was this solution helpful?