프로그래머스 - 주식가격

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42584?language=python3

프로그래머스

 

 

 

풀이

인덱스의 차이를 통해 가격이 떨어지지 않는 기간을 구한다.

주식 가격이 떨어지는 인덱스를 기준으로 다른 인덱스와의 길이를 구할 수 있다.

가격이 떨어지는 인덱스보다 이전 인덱스의 가격이 높으면 이전 인덱스의 길이를 확정한다.

스택을 이용해 인덱스를 저장해 길이 비교를 쉽게 한다.

스택에 마지막까지 남아있는 인덱스는 가격이 떨어지지 않는 주식 가격이다.

이 인덱스들은 따로 길이를 계산한다.

 

 

 

코드

python

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