[프로그래머스 | 파이썬 python] 주식가격

👀 문제

프로그래머스

 

👁 풀이

def solution(prices):
    answer = [0 for _ in range(len(prices))]
    stack = []
    for idx,price in enumerate(prices):
        while stack and stack[-1][1] > price:
            x = stack.pop()
            answer[x[0]] = idx-x[0]
        stack.append([idx,price])
        
    for i, s in stack:
        answer[i] = len(prices) - i - 1
        
    return answer