feature: best time to buy and sell stock revisit

solairerove · Nov 21, 2023 · 5b4ef5a · 5b4ef5a
1 parent 6c66601
commit 5b4ef5a
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/dynamic_programming/BestTimeToBuyAndSellStock.py b/dynamic_programming/BestTimeToBuyAndSellStock.py
@@ -3,9 +3,11 @@
 
 # O(n) time || O(1) space
 def max_profit(self, prices: List[int]) -> int:
-    min_price, max_profit = float('inf'), 0
-    for price in prices:
-        min_price = min(min_price, price)
-        max_profit = max(max_profit, price - min_price)
+    min_price, res = prices[0], 0
+    for price in prices[1:]:
+        if price < min_price:
+            min_price = price
+        else:
+            res = max(res, price - min_price)
 
-    return max_profit
+    return res