From b8111cfb2852411478dd53ada9258d97af52f767 Mon Sep 17 00:00:00 2001 From: ivan Date: Sat, 14 Dec 2024 04:20:12 -0600 Subject: [PATCH] adding best time to sell stock --- .../best_time_to_sell_stock.py | 20 +++++++++++++++++++ .../test_best_time_to_sell_stock_round_12.py | 11 ++++++++++ 2 files changed, 31 insertions(+) create mode 100644 src/my_project/interviews/top_150_questions_round_12/best_time_to_sell_stock.py create mode 100644 tests/test_150_questions_round_12/test_best_time_to_sell_stock_round_12.py diff --git a/src/my_project/interviews/top_150_questions_round_12/best_time_to_sell_stock.py b/src/my_project/interviews/top_150_questions_round_12/best_time_to_sell_stock.py new file mode 100644 index 00000000..cebffd2c --- /dev/null +++ b/src/my_project/interviews/top_150_questions_round_12/best_time_to_sell_stock.py @@ -0,0 +1,20 @@ +from typing import List, Union, Collection, Mapping, Optional +from abc import ABC, abstractmethod +import math + +class Solution: + def maxProfit(self, prices: List[int]) -> int: + + min_price = math.inf + max_profit = 0 + len_prices = len(prices) + + for i in range(len_prices): + + if prices[i] < min_price: + min_price = prices[i] + elif prices[i] - min_price > max_profit: + max_profit = prices[i] - min_price + + return max_profit + diff --git a/tests/test_150_questions_round_12/test_best_time_to_sell_stock_round_12.py b/tests/test_150_questions_round_12/test_best_time_to_sell_stock_round_12.py new file mode 100644 index 00000000..579f66fd --- /dev/null +++ b/tests/test_150_questions_round_12/test_best_time_to_sell_stock_round_12.py @@ -0,0 +1,11 @@ +import unittest +from src.my_project.interviews.top_150_questions_round_12\ +.best_time_to_sell_stock import Solution + +class BestTimeToSellStockTestCase(unittest.TestCase): + + def test_best_time_to_sell_stock(self): + solution = Solution() + output = solution.maxProfit(prices=[7,1,5,3,6,4]) + target = 5 + self.assertEqual(output, target) \ No newline at end of file