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

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

코딩테스트 연습 - 주식가격

#프로그래머스 주식가격
def solution(prices):
    answer = [0]*(len(prices))
    lst=[]
    for i in range (len(prices)):
        lst.append([i,prices[i]])
    stack=[]
    for i in range(len(prices)):
        if stack:
            k=prices[i]
            while stack and stack[-1][1]>k:
                x=stack.pop()
                answer[x[0]]=(i-x[0])
        stack.append(lst[i])
    for ind in range (len(stack)):
        answer[stack[ind][0]]=(i-stack[ind][0])
    return answer

처음에는 dp문제인줄 알고 dp로 접근했는데, 스택과 큐를 이용하는 문제였다.

 

lst라는 변수에 각 변수의 index와 값을 저장한다. 그 후, lst에서 값을 하나씩 가져와서 stack에 넣고, 가져온값이 stack에 있는 값보다 크면 stack의 마지막부터 pop 한다.