leetcode.com 122. Best Time to Buy and Sell Stock II

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

Author: skysign

leetcode.com 122. Best Time to Buy and Sell Stock II/main.py

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

leetcode.com 122. Best Time to Buy and Sell Stock II/test_main.py

@@ -0,0 +1,15 @@
+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]), 7)
+
+    def test_max_profit2(self):
+        sln = Solution()
+        self.assertEqual(sln.maxProfit([1,2,3,4,5]), 4)
+
+    def test_max_profit3(self):
+        sln = Solution()
+        self.assertEqual(sln.maxProfit([7,6,4,3,1]), 0)