🧨 문제

prices는 배열이며, 각 요소는 매일의 주식 가격이다. 한 번씩만 사고 팔 수 있다면,
제일 큰 이익은 얼마인지 계산하는 함수를 만들어라

input: [7, 1, 5, 3, 6, 4]
output: 5

1(2일)에서 사서 6(5일)에 파는 것이 가장 큰 수익
(미래인 1에서 사서, 과거 7에서 파는 것 안됌)

input: [7, 6, 4, 3, 1]
output: 0

매일 가격이 낮아지기 때문에 수익이 없음

🎯 내 풀이

def maxProfit(prices): 
#step1
    after_price = []
    result = set()

#step2
    for price in sorted(prices):
    
        if prices.index(price) != len(prices) - 1:
            after_price = prices[prices.index(price) + 1:]
            
        else:
            after_price = [price]
            
#step3
        profit = max(after_price) - price
        
        if after_price and profit >= 0:
          result.add(profit)
    
    return max(result)
    

가격을 낮은 가격순으로 정렬한 리스트를 for문으로 순회하면서, 각각에 대해 가장 큰 시세 차익을 계산한다

시세 차익이 0이상이면, 그 값을 result에 담고
가장 큰 result를 반환한다.

📍 step1

def maxProfit(prices): 
#step1
    after_price = []
    result = set()

특정 가격일 때, 그 가격 이후의 가격을 저장할 after_price, 시세 차익을 담을 result를 생성한다.

📍 step2

def maxProfit(prices): 
#... ...#

#step2
    for price in sorted(prices):
    
        if prices.index(price) != len(prices) - 1:
            after_price = prices[prices.index(price) + 1:]
            
        else:
            after_price = [price] 

정렬된 prices를 순회하며, price 뒤에 리스트를 잘라 after_price에 담는다.

만약 price가 prices 리스트의 가장 끝 요소이면, after_price를 price만을 원소로 갖는 리스트로 초기화 한다.

📍 step3

def maxProfit(prices): 
#... ...#

#... ...#
            
#step3
        profit = max(after_price) - price
        
        if after_price and profit >= 0:
          result.add(profit)
    
    return max(result)

profit 이 0보다 크거나 같다면, result에 추가하고 result의 max값을 반환한다.