Maximizing a particular sum over all possible subarrays

Question

Consider an array like this one below:

  {1, 5, 3, 5, 4, 1}

When we choose a subarray, we reduce it to the lowest number in the subarray. For example, the subarray {5, 3, 5} becomes {3, 3, 3}. Now, the sum of the subarray is defined as the sum of the resultant subarray. For example, {5, 3, 5} the sum is 3 + 3 + 3 = 9. The task is to find the largest possible sum that can be made from any subarray. For the above array, the largest sum is 12, given by the subarray {5, 3, 5, 4}.

Is it possible to solve this problem in time better than O(n²)?

Answer 1

I believe that I have an algorithm for this that runs in O(n) time. I'll first describe an unoptimized version of the algorithm, then give a fully optimized version.

For simplicity, let's initially assume that all values in the original array are distinct. This isn't true in general, but it gives a good starting point.

The key observation behind the algorithm is the following. Find the smallest element in the array, then split the array into three parts - all elements to the left of the minimum, the minimum element itself, and all elements to the right of the minimum. Schematically, this would look something like

 +-----------------------+-----+-----------------------+
 |     left values       | min |      right values     |
 +-----------------------+-----+-----------------------+     

Here's the key observation: if you take the subarray that gives the optimum value, one of three things must be true:

1. That array consists of all the values in the array, including the minimum value. This has total value min * n, where n is the number of elements.
2. That array does not include the minimum element. In that case, the subarray has to be purely to the left or to the right of the minimum value and cannot include the minimum value itself.

This gives a nice initial recursive algorithm for solving this problem:

• If the sequence is empty, the answer is 0.
• If the sequence is nonempty:
  • Find the minimum value in the sequence.
  • Return the maximum of the following:
    • The best answer for the subarray to the left of the minimum.
    • The best answer for the subarray to the right of the minimum.
    • The number of elements times the minimum.

So how efficient is this algorithm? Well, that really depends on where the minimum elements are. If you think about it, we do linear work to find the minimum, then divide the problem into two subproblems and recurse on each. This is the exact same recurrence you get when considering quicksort. This means that in the best case it will take Θ(n log n) time (if we always have the minimum element in the middle of each half), but in the worst case it will take Θ(n²) time (if we always have the minimum value purely on the far left or the far right.

Notice, however, that all of the effort we're spending is being used to find the minimum value in each of the subarrays, which takes O(k) time for k elements. What if we could speed this up to O(1) time? In that case, our algorithm would do a lot less work. More specifically, it would do only O(n) work. The reason for this is the following: each time we make a recursive call, we do O(1) work to find the minimum element, then remove that element from the array and recursively process the remaining pieces. Each element can therefore be the minimum element of at most one of the recursive calls, and so the total number of recursive calls can't be any greater than the number of elements. This means that we make at most O(n) calls that each do O(1) work, which gives a total of O(1) work.

So how exactly do we get this magical speedup? This is where we get to use a surprisingly versatile and underappreciated data structure called the Cartesian tree. A Cartesian tree is a binary tree created out of a sequence of elements that has the following properties:

• Each node is smaller than its children, and
• An inorder walk of the Cartesian tree gives back the elements of the sequence in the order in which they appear.

For example, the sequence 4 6 7 1 5 0 2 8 3 has this Cartesian tree:

       0
      / \
     1   2
    / \   \
   4   5   3
    \     /
     6   8
      \
       7

And here's where we get the magic. We can immediately find the minimum element of the sequence by just looking at the root of the Cartesian tree - that takes only O(1) time. Once we've done that, when we make our recursive calls and look at all the elements to the left of or to the right of the minimum element, we're just recursively descending into the left and right subtrees of the root node, which means that we can read off the minimum elements of those subarrays in O(1) time each. Nifty!

The real beauty is that it is possible to construct a Cartesian tree for a sequence of n elements in O(n) time. This algorithm is detailed in this section of the Wikipedia article. This means that we can get a super fast algorithm for solving your original problem as follows:

• Construct a Cartesian tree for the array.
• Use the above recursive algorithm, but use the Cartesian tree to find the minimum element rather than doing a linear scan each time.

Overall, this takes O(n) time and uses O(n) space, which is a time improvement over the O(n²) algorithm you had initially.

At the start of this discussion, I made the assumption that all array elements are distinct, but this isn't really necessary. You can still build a Cartesian tree for an array with non-distinct elements in it by changing the requirement that each node is smaller than its children to be that each node is no bigger than its children. This doesn't affect the correctness of the algorithm or its runtime; I'll leave that as the proverbial "exercise to the reader." :-)

This was a cool problem! I hope this helps!

Answer 2

Assuming that the numbers are all non-negative, isn't this just the "maximize the rectangle area in a histogram" problem? which has now become famous...

O(n) solutions are possible. This site: http://blog.csdn.net/arbuckle/article/details/710988 has a bunch of neat solutions.

To elaborate what I am thinking (it might be incorrect) think of each number as histogram rectangle of width 1.

By "minimizing" a subarray [i,j] and adding up, you are basically getting the area of the rectangle in the histogram which spans from i to j.

This has appeared before on SO: Maximize the rectangular area under Histogram, you find code and explanation, and a link to the official solutions page (http://www.informatik.uni-ulm.de/acm/Locals/2003/html/judge.html).

Answer 3

The following algorithm I tried will have the order of the algorithm which is initially used to sort the array. For example, if the initial array is sorted with binary tree sort, it will have O(n) in best case and O(n log n) as an average case.

Gist of algorithm:

The array is sorted. The sorted values and the correponding old indices are stored. A binary search tree is created from the corresponding older indices which is used to determine how far it can go forwards and backwards without encountering a value less than the current value, which will result in the maximum possible sub array.

I will explain the method with the array in the question [1, 5, 3, 5, 4, 1]

                      1  5  3  5  4  1
                  -------------------------
 array indices =>     0  1  2  3  4  5  
                  -------------------------

This array is sorted. Store the value and their indices in ascending order, which will be as follows

                                   1  1  3  4  5  5
                                 -------------------------
 original array indices =>         0  5  2  4  1  3  
 (referred as old_index)         -------------------------

It is important to have a reference to both the value and their old indices; like an associative array;

Few terms to be clear:

old_index refers to the corresponding original index of an element (that is index in original array);

For example, for element 4, old_index is 4; current_index is 3;

whereas, current_index refers to the index of the element in the sorted array; current_array_value refers to the current element value in the sorted array.

pre refers to inorder predecessor; succ refers to inorder successor

Also, min and max values can be got directly, from first and last elements of the sorted array, which are min_value and max_value respectively;

Now, the algorithm is as follows which should be performed on sorted array.

Algorithm:

Proceed from the left most element.

For each element from the left of the sorted array, apply this algorithm

    if(element == min_value){

    max_sum = element * array_length;

        if(max_sum > current_max)
        current_max = max_sum;

        push current index into the BST;

    }else if(element == max_value){

        //here current index is the index in the sorted array
        max_sum = element * (array_length - current_index);

        if(max_sum > current_max)
        current_max = max_sum;


        push current index into the BST;

    }else {

        //pseudo code steps to determine maximum possible sub array with the current element 

        //pre is inorder predecessor and succ is inorder successor

        get the inorder predecessor and successor from the BST;



        if(pre == NULL){

            max_sum = succ * current_array_value;


            if(max_sum > current_max)
            current_max = max_sum;


        }else if (succ == NULL){

            max_sum = (array_length - pre) - 1) * current_array_value;

            if(max_sum > current_max)
            current_sum = max_sum;

        }else {

        //find the maximum possible sub array streak from the values

        max_sum = [((succ - old_index) - 1) + ((old_index - pre) - 1) + 1] * current_array_value;

            if(max_sum > current_max)
            current_max = max_sum;

        } 

    }

For example,

original array is

                      1  5  3  5  4  1
                  -------------------------
 array indices =>     0  1  2  3  4  5  
                  -------------------------

and the sorted array is

                                   1  1  3  4  5  5
                                 -------------------------
 original array indices =>         0  5  2  4  1  3  
 (referred as old_index)         -------------------------

After first element:

max_sum = 6 [it will reduce to 1*6]

        0

After second element:

max_sum = 6 [it will reduce to 1*6]

        0
         \
          5

---

After third element:

        0
         \
          5
         /
        2

inorder traversal results in: 0 2 5

applying the algorithm,

max_sum = [((succ - old_index) - 1) + ((old_index - pre) - 1) + 1] * current_array_value;

max_sum = [((5-2)-1) + ((2-0)-1) + 1] * 3 = 12

current_max = 12 [the maximum possible value]

---

After fourth element:

        0
         \
          5
         / 
        2   
         \
          4

inorder traversal results in: 0 2 4 5

applying the algorithm,

max_sum = 8 [which is discarded since it is less than 12]

After fifth element:

max_sum = 10 [reduces to 2 * 5, discarded since it is less than 8]

After last element:

max_sum = 5 [reduces to 1 * 5, discarded since it is less than 8]

This algorithm will have the order of the algorithm which is initially used to sort the array. For example, if the initial array is sorted with binary sort, it will have O(n) in best case and O(n log n) as an average case.

The space complexity will be O(3n) [O(n + n + n), n for sorted values, another n for old indices, and another n for constructing the BST]. However, I'm not sure about this. Any feedback on the algorithm is appreciated.