[코딩테스트] 프로그래머스 주식가격 파이썬(Python)

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

프로그래머스

from collections import deque
def solution(prices):
    q = deque(prices)
    result = []
    while q:
        c = q.popleft()
        i = 1
        for p in q:
            if p < c:
                result.append(i)
                break
            i+=1
        else:
            result.append(len(q))
    return result

먼저 저는 collections 모듈의 deque자료구조를 이용하여 prices를 deque자료구조의 q로 바꾸었습니다.

 

q가 빌때까지 반복합니다.
먼저 q.popleft()를 하여 q의 가장 첫번째 요소인 c를 추출합니다.

• c는 현재 시점의 가격을 의미합니다.

i = 1 로 시간에 해당하고 점점 증가합니다.

 

다음은 for문으로 나머지 q를 반복하면서
만약 현재시점의 가격(c) 보다 p의 값이 작다면 즉 주식가격이 떨어졌다면 시간(i)를 result에 저장하고
break를 사용하여 반복문을 빠져나옵니다.
만약 주식 가격이 떨어지지 않았다면 나머지 값들 중 가격이 떨어진 값이 있는지 확인해야 하기 때문에 나머지 요소를 확인합니다

• 이때 i+=1로 시간을 증가시킵니다.

하지만 break에서 반복문을 빠져나오지 않고 모든 for문을 돌았다면 주식 가격이 끝까지 가격이 떨어지지 않았음을 의미합니다.
따라서 len(q)를 result에 저장합니다.

모든 시점의 주식 가격을 확인했다면 result를 반환해줍니다.