Python/알고리즘

[파이썬 알고리즘 인터뷰] 12. 주식을 사고 팔기 가장 좋은 시점

AI 꿈나무 2021. 2. 13. 16:32
반응형

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 코드 출처 : 파이썬 알고리즘 인터뷰 깃허브

 


12. 주식을 사고팔기 가장 좋은 시점

리트코드 121. Best Time to Buy and Sell Stock 문제입니다.

leetcode.com/problems/best-time-to-buy-and-sell-stock/

 

 

풀이

 price의 최소값을 계속 갱신하면서, 가격 차이를 계산하여 차이가 클 경우 최대값을 저장하는 풀이입니다.

 

def maxProfit(self, prices):
    profit = 0
    min_price = sys.maxsize

    # 최소값과 최댓값을 계속 생신
    for price in prices:
        min_price = min(min_price, price)
        profit = max(profit, price - min_price)

    return profit
반응형