[프로그래머스] 주식가격 / Python / 스택&큐

주식가격

문제 설명

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

제한사항

• prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
• prices의 길이는 2 이상 100,000 이하입니다.

풀이 방법

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

    for i, x in enumerate(prices):
    	# 가격이 1이라면 끝까지 가격이 떨어지지 않음
        if x == 1:
            answer[i] = len(prices) - i - 1
            continue
        
        # 다음 가격을 확인하면서 가격이 떨어지지 않은 기간 갱신
        j = i
        while j < len(prices) - 1:
            answer[i] += 1
            if x > prices[j + 1]:
                break
            j += 1

    return answer