Add more test cases for finding max sub-array algorithm.

This commit is contained in:
Oleksii Trekhleb
2018-09-04 11:35:13 +03:00
parent 2a2b5daa7d
commit 814fa773ca
3 changed files with 21 additions and 14 deletions

View File

@@ -6,36 +6,35 @@
* @return {Number[]}
*/
export default function dpMaximumSubarray(inputArray) {
// We iterate through the inputArray once, using a greedy approach
// to keep track of the maximum sum we've seen so far and the current sum
// We iterate through the inputArray once, using a greedy approach to keep track of the maximum
// sum we've seen so far and the current sum.
//
// currentSum gets reset to 0 everytime it drops below 0
// The currentSum variable gets reset to 0 every time it drops below 0.
//
// maxSum is set to -Infinity so that if all numbers
// are negative, the highest negative number will constitute
// the maximum subarray
// The maxSum variable is set to -Infinity so that if all numbers are negative, the highest
// negative number will constitute the maximum subarray.
let maxSum = -Infinity;
let currentSum = 0;
// We need to keep track of the starting and ending indices that
// contributed to our maxSum so that we can return the actual subarray
// We need to keep track of the starting and ending indices that contributed to our maxSum
// so that we can return the actual subarray.
let maxStartIndex = 0;
let maxEndIndex = inputArray.length;
let currentStartIndex = 0;
inputArray.forEach((num, currentIndex) => {
currentSum += num;
inputArray.forEach((currentNumber, currentIndex) => {
currentSum += currentNumber;
// Update maxSum and the corresponding indices
// if we have found a new max
// Update maxSum and the corresponding indices if we have found a new max.
if (maxSum < currentSum) {
maxSum = currentSum;
maxStartIndex = currentStartIndex;
maxEndIndex = currentIndex + 1;
}
// Reset currentSum and currentStartIndex
// if currentSum drops below 0
// Reset currentSum and currentStartIndex if currentSum drops below 0.
if (currentSum < 0) {
currentSum = 0;
currentStartIndex = currentIndex + 1;