[Leetcode/파이썬] 121. Best Time to Buy and Sell Stock

Best Time to Buy and Sell Stock

---

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.

 

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 <= 105
• 0 <= prices[i] <= 104

 

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [0] * len(prices)
        curr = prices[0]
        for i in range(1, len(prices)) :
            if curr > prices[i] :
                curr = prices[i]
            else :
                dp[i] = prices[i] - curr

        return max(dp)

dp 문제. 이렇게 풀어도? 잘 되는거 보고 좀 놀랐다. 

DP 테이블 하나 생성하고, 현재값을 prices[0]으로 잡았다.

현재값을 기준으로

1. 다음값이 더 낮은 값이 오면 현재 값을 다음 값으로 교체. 
2. 아니라면 다음 값 - 현재 값(이때 현재값은 유지)

를 반복한다. 이렇다 보니 현재값은 가장 낮은 값을 찾게 되고 그 다음에 등장하는 값 중 가장 큰 격차가 남는 값을 return 하게 된다.