arraysoptimizationcomplexity-theorystocks

Find buy/sell prices in array of stock values to maximize positive difference


Got this question in an interview today, and its optimized solution stopped me cold (which blows, because I really wanted to work for this company...)

Given a single array of real values, each of which represents the stock value of a company after an arbitrary period of time, find the best buy price and its corresponding best sell price (buy low, sell high).

To illustrate with an example, let's take the stock ticker of Company Z:

55.39 109.23 48.29 81.59 105.53 94.45 12.24

Important to note is the fact that the array is "sorted" temporally - i.e. as time passes, values are appended to the right end of the array. Thus, our buy value will be (has to be) to the left of our sell value.

(in the above example, the ideal solution is to buy at 48.29 and sell at 105.53)

I came up with the naive solution easily enough with O(n2) complexity (implemented in java):

// returns a 2-element array: first element is the index in the argument array
// of the best buying price, and the second element is the index of the best
// selling price which, collectively, maximize the trading return
//
// if there is no favorable trading (e.g. prices monotonically fall), null is returned
public int[] maximizeReturn(ArrayList<Double> prices) {
  int [] retval = new int[2];
  int BUY = 0, SELL = 1;
  retval[BUY] = retval[SELL] = -1; // indices of buy and sell prices, respectively

  for (int i = 0; i < prices.size(); i++) {
    for (int j = i + 1; j < prices.size(); j++) {
      double difference = prices.get(j).doubleValue() - 
                          prices.get(i).doubleValue();

      if (difference > 0.0) {
        if (retval[BUY] < 0 || difference > prices.get(retval[SELL]).doubleValue() - 
                                            prices.get(retval[BUY]).doubleValue()) {
          retval[BUY] = i;
          retval[SELL] = j;
        }
      }
    }
  }
  return (retval[BUY] > 0 ? retval : null);
}

Here's where I screwed up: there's a linear time O(n) solution, and I completely bombed in trying to figure it out (yeah, I know, FAIL). Does anyone know how to implement the linear time solution? (any language you're comfortable with) Thanks!

Edit

I suppose, for anyone interested, I just received word today that I didn't get the job for which I interviewed where they asked me this question. :(


Solution

  • Here's an attempt (C++). Basically everytime I track a new top, I try to see if thats the best profit thusfar. I know that the "bottom" must have been discovered earlier. At that point I remember the top, bottom, and the current max profit. If a new bottom is discovered later, its AFTER the current top, so we must reset top and see if a slightly lower "top" can yield better profit.

    #include <iostream>
    
    int main()
    {
    
        double REALLY_BIG_NO = 1e99;
        double bottom = REALLY_BIG_NO; // arbirtrary large number
        double currBestBuy = 0.0;
        double top = 0.0;
        double currBestSell = 0.0;
        double profit = 0.0;
    
        // array of prices
        double prices[] = {10.50, 55.39, 109.23, 48.29, 81.59, 105.53, 94.45, 12.24, 152.0, 2, 170.0};
        int numPrices = 10;// number of prices
    
        for (int i = 0; i < numPrices; ++i)
        {
             if (prices[i] < bottom)
             {
                bottom = prices[i];
                // reset the search on a new bottom
                top = 0.0;
             }
             else if (prices[i] > top)
             {
                top = prices[i];
               // calculate profit
                double potentialProfit = (top - bottom);
                if (potentialProfit > profit &&
                    bottom != REALLY_BIG_NO)
                {
                    profit = potentialProfit;
                    currBestSell = top;
                    currBestBuy = bottom;
                }
             }
        }
    
        std::cout << "Best Buy: " << currBestBuy << "Best Sell: " << currBestSell << std::endl;
    }
    

    So far I've played around with a bunch of different input sets, and so far I haven't had any problems... (let me know if you test this and see anything wrong)

    I highly recommend using Austin Salonen's updated answer to this question and adapting it to your language.