[프로그래머스] 주식가격

1. 문제

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

2. 접근 방법

스택을 사용해서 접근했습니다

 

3. 코드

Python

def solution(prices):
    N = len(prices)
    answer = [0]*N
    stack = []

    for idx, price in enumerate(prices):
        while stack and prices[stack[-1]] > price:
            j = stack.pop()
            answer[j] = idx-j
        stack.append(idx)
    
    for i in stack:
        answer[i] = N-i-1
    
    return answer

4. 마치며