about summary refs log blame commit diff
path: root/users/wpcarro/scratch/facebook/stock-price.py
blob: 8e42f815231178cbf8bd499906a7d8b6ebd0e866 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16















                                         
def max_profit(xs):
    buy = xs[0]
    profit = xs[1] - xs[0]
    for price in xs[1:]:
        profit = max(profit, price - buy)
        buy = min(buy, price)
    return profit

xs = [([10,7,5,8,11,9], 6),
      ([10,8,7,6,5], -1)]

for x, expected in xs:
    result = max_profit(x)
    print(x, result)
    assert result == expected
    print("Success!")