-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path121. Best time to buy & sell stock .py
37 lines (28 loc) · 1.12 KB
/
121. Best time to buy & sell stock .py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
'''
You are given an array prices where prices[i] is the price of a given stock on the ith day.
You want to maximize your profit by choosing a single day to buy one stock and choosing a different day in the future to sell that stock.
Return the maximum profit you can achieve from this transaction. If you cannot achieve any profit, return 0.
Example 1:
Input: prices = [7,1,5,3,6,4]
Output: 5
Explanation: Buy on day 2 (price = 1) and sell on day 5 (price = 6), profit = 6-1 = 5.
Note that buying on day 2 and selling on day 1 is not allowed because you must buy before you sell.
Example 2:
Input: prices = [7,6,4,3,1]
Output: 0
Explanation: In this case, no transactions are done and the max profit = 0.'''
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if len(prices) == 1:
return 0
else:
profit = 0
s = 0
e = 1
while e < len(prices):
if prices[e] > prices[s]:
profit = max(profit, prices[e] - prices[s])
else:
s = e
e += 1
return profit