[Python] 주식가격

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

풀이

1. 큐를 사용

2. 맨 앞쪽 것을 빼고 큐에 남은 것들을 비교하는동안 sec 증가

3. 맨 앞쪽 것보다 큐에 남은 것이 값이 작다면 break

 

from collections import deque

def solution(prices):
    answer = []
    queue = deque(prices)
    
    while queue:
        price = queue.popleft()
        sec = 0
        for q in queue:
            sec += 1
            if price > q:
                break
        answer.append(sec)
    
    return answer