leetcode.com 121. Best Time to Buy and Sell Stock

문제 링크: https://leetcode.com/problems/best-time-to-buy-and-sell-stock/description/

Author: skysign

leetcode.com 121. Best Time to Buy and Sell Stock v2/main.py

@@ -0,0 +1,15 @@
+from typing import List
+
+
+class Solution:
+    def maxProfit(self, prices: List[int]) -> int:
+        answer = 0
+        buy = prices[0]
+
+        for price in prices[1:]:
+            if buy >= price:
+                buy = price
+            else:
+                answer = max(answer, price - buy)
+
+        return answer

leetcode.com 121. Best Time to Buy and Sell Stock v2/test_main.py

@@ -0,0 +1,12 @@
+from unittest import TestCase
+from main import Solution
+
+
+class TestSolution(TestCase):
+    def test_max_profit1(self):
+        sln = Solution()
+        self.assertEqual(sln.maxProfit([7,1,5,3,6,4]), 5)
+
+    def test_max_profit2(self):
+        sln = Solution()
+        self.assertEqual(sln.maxProfit([7,6,4,3,1]), 0)