Apple stocks interview question

Written by vadim-samokhin | Published 2018/01/05
Tech Story Tags: algorithms | interview-questions | coding-interviews | software-development

TLDRvia the TL;DR App

Recently I stumbled upon this interview question on InterviewCake. Basically it goes like this: you have an array of stock prices, and you should calculate the highest profit you could get. For example, you have [2, 1, 4, 10, 7] — the highest profit is 10–1 = 9. What I didn’t like about the problem was that it didn’t require to find minimum and maximum stock prices you could obtain the best profit with, just the highest profit itself. So I decided to solve it with this clause. Besides, I believe it requires to delve into the nature of this problem a bit deeper.

After hanging around and re-reading the problem for I guess half an hour, hoping to come up with a solution, I decided to draw the diagram of how the stock prices evolve.

Here is the simplest beginning:

The best profit I could obtain so far is buy for $10 and sell for $20. How could this graph evolve? Well, there are actually only two scenarios that I’m interested in. If the next price is higher than the previous, I have a new highest price, and new highest profit. And if the next price is lower than the previous lowest price, I potentially have a lowest price that I should buy stock for. But I don’t know if later there will be a price that results in a new highest profit. It could be like that:

or like that:

So I need to introduce a concept of minimum price candidate, that is, a price that could be the one I should buy the stocks for. So when I find myself in point 3, I say that it is a candidate for the next minimum price. And if the stock at point 4 leads to higher profit (or, in other words, the highest so far), I pose that the candidate for minimum price actually becomes a minimum price, and the stock that brings me to the highest profit becomes the maximum price, that is, the price I should sell the stocks for.

So the following code seems fine from the first sight:

function getMaxProfit(array $stocks){$minCandidate = $min = $stocks[0];$max = $stocks[1];

**for** ($i = 1; $i < _count_($stocks); $i++) {  
    **if** ($stocks\[$i\] < $minCandidate) {  
        $minCandidate = $stocks\[$i\];  
    } **elseif** ($stocks\[$i\] - $minCandidate > $max - $min) {  
        $max = $stocks\[$i\];  
        $min = $minCandidate;  
    }  
}  

**return** \[$min, $max\];  

}

But there is a problem. What if stock prices go down the whole day, but the difference between the prices gets lower? For example, with this array — [10, 5, 2, 1] — I get an erroneous result. So I need to calculate the highest profit first, and then refresh $minCandidate if needed:

function getMaxProfit(array $stocks){$minCandidate = $min = $stocks[0];$max = $stocks[1];

**for** ($i = 1; $i < _count_($stocks); $i++) {  
    **if** ($stocks\[$i\] - $minCandidate > $max - $min) {  
        $max = $stocks\[$i\];  
        $min = $minCandidate;  
    }  

    **if** ($stocks\[$i\] < $minCandidate) {  
        $minCandidate = $stocks\[$i\];  
    }  
}  

**return** \[$min, $max\];  

}

That’s it.


Written by vadim-samokhin | Aspiring object-thinker
Published by HackerNoon on 2018/01/05