Maximum Sum of Three Non-Overlapping Subarrays II in JavaScript (Time Complexity: O(n))
Given an array nums of integers, find three non-overlapping subarrays with maximum sum.
Return the total sum of the three subarrays
Example:
Input: [2, 3, -8, 7, -2, 9, -9, 7, -2, 4]
Output: 28
Explanation: Subarrays [2, 3], [7, -2, 9] and [7, -2, 4]
have the maximum sum of 28
Note:
- Subarrays must be non-empty
- nums contains at least three numbers
Understanding the Problem
The core challenge of this problem is to find three non-overlapping subarrays that together have the maximum possible sum. This problem is significant in scenarios where we need to maximize the sum of multiple segments of data, such as in financial analysis or signal processing.
Potential pitfalls include overlapping subarrays and not considering all possible subarray combinations.
Approach
To solve this problem, we can break it down into smaller steps:
- Calculate the sum of all possible subarrays of a fixed length.
- Use dynamic programming to keep track of the best subarray sums up to each point in the array.
- Combine the results to find the maximum sum of three non-overlapping subarrays.
Naive Solution
A naive solution would involve checking all possible combinations of three subarrays, but this approach is not optimal due to its high time complexity.
Optimized Solution
We can optimize the solution using dynamic programming and prefix sums. The idea is to precompute the sums of all subarrays of a given length and then use these precomputed sums to find the maximum sum of three non-overlapping subarrays.
Algorithm
Here is a step-by-step breakdown of the optimized algorithm:
- Compute the prefix sums of the array to quickly calculate subarray sums.
- Use three arrays to store the best subarray sums up to each point in the array for the first, second, and third subarrays.
- Iterate through the array to update these arrays with the maximum sums found so far.
- Combine the results to find the maximum sum of three non-overlapping subarrays.
Code Implementation
// Function to find the maximum sum of three non-overlapping subarrays
function maxSumOfThreeSubarrays(nums) {
const n = nums.length;
const k = 3; // Length of each subarray
const sum = new Array(n + 1).fill(0);
// Compute prefix sums
for (let i = 0; i < n; i++) {
sum[i + 1] = sum[i] + nums[i];
}
// Arrays to store the best subarray sums up to each point
const left = new Array(n).fill(0);
const right = new Array(n).fill(0);
// Calculate the best subarray sums for the left side
let total = sum[k] - sum[0];
for (let i = k; i < n; i++) {
if (sum[i + 1] - sum[i + 1 - k] > total) {
total = sum[i + 1] - sum[i + 1 - k];
left[i] = i + 1 - k;
} else {
left[i] = left[i - 1];
}
}
// Calculate the best subarray sums for the right side
total = sum[n] - sum[n - k];
right[n - k] = n - k;
for (let i = n - k - 1; i >= 0; i--) {
if (sum[i + k] - sum[i] >= total) {
total = sum[i + k] - sum[i];
right[i] = i;
} else {
right[i] = right[i + 1];
}
}
// Find the maximum sum by combining the results
let maxSum = 0;
for (let i = k; i <= n - 2 * k; i++) {
const l = left[i - 1];
const r = right[i + k];
const currentSum = (sum[i + k] - sum[i]) + (sum[l + k] - sum[l]) + (sum[r + k] - sum[r]);
if (currentSum > maxSum) {
maxSum = currentSum;
}
}
return maxSum;
}
// Example usage
const nums = [2, 3, -8, 7, -2, 9, -9, 7, -2, 4];
console.log(maxSumOfThreeSubarrays(nums)); // Output: 28
Complexity Analysis
The time complexity of this solution is O(n), where n is the length of the input array. This is because we make a constant number of passes through the array. The space complexity is also O(n) due to the additional arrays used for prefix sums and tracking the best subarray sums.
Edge Cases
Potential edge cases include:
- Arrays with negative numbers.
- Arrays where the best subarrays are at the beginning or end of the array.
These edge cases are handled by the algorithm as it considers all possible subarray positions.
Testing
To test the solution comprehensively, consider the following test cases:
- Simple cases with small arrays.
- Arrays with all positive or all negative numbers.
- Arrays with mixed positive and negative numbers.
Use a testing framework like Jest or Mocha to automate the testing process.
Thinking and Problem-Solving Tips
When approaching such problems, consider breaking them down into smaller subproblems and using dynamic programming to optimize the solution. Practice solving similar problems to improve your problem-solving skills.
Conclusion
Understanding and solving problems like this one is crucial for developing strong algorithmic thinking. Practice regularly and explore different approaches to enhance your skills.
Additional Resources
For further reading and practice, consider the following resources: