프로그래머스: 주식가격

문제

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

 

문제 풀이

 초 단위로 기록된 주식 가격이 떨어지지 않는 기간을 반환하는 문제이다. 이를 위해 prices에서 첫 번째 값부터 시작하여, 다른 값과 비교하였을 때 어느 시점에서 가격이 하락하는지를 파악하면 된다. 문제 풀이를 위해 큐를 사용하여도 되고, 단순히 반복문을 통해 문제를 해결할 수도 있다.

 

• 입력되는 주식가격을 `deque`로 변환한다.
• 첫 번째 값을 가져와, 주식 가격들과 비교하며 언제까지 상승하는지 카운트한다.
• 카운트된 값을 `list`에 기록하여 반환한다.

 

코드

from collections import deque

def solution(prices):
    answer = []
    prices = deque(prices)

    while prices:
        cnt = 0
        value = prices.popleft()
        for price in prices:
            cnt += 1
            if value > price:
                break
        answer.append(cnt)
    return answer