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

https://programmers.co.kr/learn/courses/30/lessons/42584

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

입출력 예

prices                                                                      return

[1, 2, 3, 2, 3]

[4, 3, 1, 1, 0]

가격이 몇초 후에 떨어지는지 구하는 문제다.

 

prices에서 왼쪽값을 하나씩 빼서 나머지 리스트중에서 더 작은 값이 언제 나오는지 확인하면 된다.

 

[1, 2, 3, 2, 3] 중에서 1을 뺐다면 [2, 3, 2, 3]을 처음부터 비교해서 작은 값이 나오면 몇 초가 지났는지 구하고 break 해야한다.

 

정답코드

from collections import deque

def solution(prices):
    prices = deque(prices) # deque를 사용해 큐로 만듬
    answer = [] # 정답을 저장할 리스트
    while prices: # prices가 없어질 때 까지
      popped = prices.popleft() # prices에서 왼쪽값을 pop
      for i, v in enumerate(prices): # 나머지 배열 순회
        if v < popped: # pop한 값 보다 작은 값이 있다면
          answer.append(i + 1) # 몇번째 원소인지 + 1 만큼 걸림
          break
        if i == len(prices) - 1: # 마지막 값까지 왔다면 
          answer.append(len(prices)) # prices 길이만큼 걸림

    return answer + [0]