[문제 풀이 6] 프로그래머스 - 주식 가격

주식 가격

 

• 문제 유형 : Stack
• 사용언어 : Python
• 난이도 : LV. 2
• 출처 : 프로그래머스

 

https://school.programmers.co.kr/learn/courses/30/lessons/42584

프로그래머스

 

def solution(prices):
    n = len(prices)
    answer = [0]*n
    
    stack = [0]
    
    for i in range(1, n):
        while stack and prices[i] < prices[stack[-1]]:
            j = stack.pop()
            answer[j] = i-j
        stack.append(i)
    
    while stack:
        j= stack.pop()
        answer[j] = n-1-j
    
    return answer