[Python-Study Leetcode 문제풀이] 121. Best Time to Buy and Sell Stock (Easy)

You are given an array prices where prices[i] is the price of a given stock on the ith day. You want to maximize your profit by choosing a single day to buy one stock and choosing a different day in the future to sell that stock. Return the maximum profit you can achieve from this transaction. If you cannot achieve any profit, return 0. (Easy) Example 1: Input: prices = [7,1,5,3,6,4] Output: 5 Explanation: Buy on day 2 (price = 1) and sell on day 5 (price = 6), profit = 6-1 = 5. Note that buying on day 2 and selling on day 1 is not allowed because you must buy before you sell. Example 2: Input: prices = [7,6,4,3,1] Output: 0 Explanation: In this case, no transactions are done and the max profit = 0. Constraints: 1 <= prices.length <= 10^5 0 <= prices[i] <= 10^4

(문제 설명)

1. 주식의 최저점~최고점을 찾아서, 최대의 수익(profit)을 내면 되는 간단한 문제

 

2. Example2에서 처럼 계속 하한가를 향해 가면, 수익을 낼수 없으므로 profit은 0이 됨

→ profit의 최초 선언 값은 0이 되어야함

 

3. constraints에서, 주식의 값은 최대 10000이므로, 주식의 min_price는 10000으로 설정을 하고 그보다 낮은 값일시 swap하는 형태로 문제 풀이

 

 

 

 

 

In [1]:

class Solution():
    def maxProfit(self, prices:list)->int:
        
        price_min = 10000 # 최저 주식값
        profit = 0 # 수익
        
        for price_current in prices:
            # 현재 가격 - 최저값의 값과 수익의 값 중, 최대값을 수익값으로 보고 스왑함
            price_min = min(price_current, price_min)
            
            # 현재 가격 - 최저값의 값과 수익의 값 중, 최대값을 수익값으로 보고 스왑함
            profit = max(profit, price_current-price_min) 
        return profit

In [2]:

# test

prices = [7, 1, 5, 3, 6, 4]
Solution().maxProfit(prices)

Out[2]:

5