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

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

문제 설명

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

제한 사항

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

입출력 예시

입출력 예 설명

• 1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
• 2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
• 3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
• 4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
• 5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.

풀이 과정

prices의 왼쪽부터 꺼내어 검사해야하므로 deque 를 이용하는 문제였다. prices에서 하나씩 popleft 하여, for문으로 prices를 검사하며 꺼낸 price보다 작은것이 있는지 검사한 뒤 for문을 돌린 만큼 answer에 더해주면 되는 문제라 비교적 쉽고 간단했다. 이때 while문과 for문이 중첩되어 있어 효율성테스트에서 통과 못하지 않을까 생각했지만 쉽게 통과하여 오잉? 했던 문제였다.

 

from collections import deque


def solution(prices):
    answer = []
    prices = deque(prices)
    while prices:
        timer = 0
        price = prices.popleft()
        for price_ck in prices:
            timer += 1
            if price_ck < price:
                break
        answer.append(timer)

    return answer


if __name__ == "__main__":
    print(solution([1, 2, 3, 2, 3]))