"Best Time to Buy and Sell Stock II" solution

Author: lancelote

src/best_time_to_buy_and_sell_stock_ii.py

@@ -0,0 +1,14 @@
+class Solution:
+    def maxProfit(self, prices: list[int]) -> int:
+        assert prices
+
+        n = len(prices)
+        total_profit = 0
+        last = prices[-1]
+
+        for i in range(n - 1, -1, -1):
+            if prices[i] < last:
+                total_profit += last - prices[i]
+            last = prices[i]
+
+        return total_profit

tests/test_best_time_to_buy_and_sell_stock_ii.py

@@ -0,0 +1,15 @@
+import pytest
+
+from src.best_time_to_buy_and_sell_stock_ii import Solution
+
+
+@pytest.mark.parametrize(
+    "prices,expected_profit",
+    (
+        ([7, 1, 5, 3, 6, 4], 7),
+        ([1, 2, 3, 4, 5], 4),
+        ([7, 6, 4, 3, 1], 0),
+    ),
+)
+def test_solution(prices, expected_profit):
+    assert Solution().maxProfit(prices) == expected_profit