[프로그래머스] LEVEL2 주식가격 파이썬

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

---

✔️문제

문제 설명

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

제한사항

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

입출력 예

prices	return
[1, 2, 3, 2, 3]	[4, 3, 1, 1, 0]

---

✔️풀이

내 풀이

def solution(prices):
    answer = []
    while len(prices):
        now = prices.pop(0)
        time = 0
        for x in prices:
            time += 1
            if x<now :
                break
            
        answer.append(time)
        
    return answer

정확성 테스트는 모두 통과했지만 효율성 테스트에서 모두 시간 초과로 실패한 코드. pop(0)의 시간 복잡도가 O(N)이라 deque를 사용하면 좋다고 배운 것이 생각나 코드를 수정했다.

from collections import deque
def solution(prices):
    answer = []
    prices = deque(prices)
    while len(prices):
        now = prices.popleft()
        time = 0
        for x in prices:
            time += 1
            if x<now :
                break
            
        answer.append(time)
        
    return answer