stack&queueLv2주식가격

문제

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

제한사항
prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
prices의 길이는 2 이상 100,000 이하입니다.
입출력 예
prices return
[1, 2, 3, 2, 3][4, 3, 1, 1, 0]
입출력 예 설명
1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.

풀이

• 배열을 deque로 바꾸어 popleft()하여 기준을 삼는다.
• 그리고 for문을 돌면서 더 작은 수가 등장하기 전까진 cnt값을 +1 누적시킨다

코드

from collections import deque 

def solution(prices):
    answer = []
    prices= deque(prices)
    while prices:
        cnt = 0
        tmp = prices.popleft()
        for x in prices:
            cnt += 1
            if tmp > x:
                break
        answer.append(cnt)
    return answer

배운 것

이전 코드를 비슷하게 작성했지만 시간초과가 났다.

def solution(prices):
    answer = []
    while prices:
        cnt = 0
        tmp = prices.pop(0)
        for x in prices:
            if x >= tmp:
                cnt += 1
        answer.append(cnt)
    return answer

비슷하지만, deque를 안썼다는 점과 if를 거쳐 하나의 명령을 더 실행하는 부분에 있어서 시간초과가 났다. if 문은 break를 잡아낼때나 사용하자.