[프로그래머스] [Level 2] [스택/큐] 주식가격

주식 가격

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

 

• 내 풀이 1
  • 테스트 케이스 1개 빼고 fail
  • 바로 떨어질 경우만 1초로 간주하는 줄 알았음
    • → 떨어지는 경우까지 버틴 시간을 구해야 하기 때문에 if count~ 문이 없어야 함

from collections import deque
def solution(prices):
    queue=deque(prices)
    answer = []
        
    while queue:
        count=0
        q=queue.popleft()
        for a in queue:
            if a<q:
                if count==0:
                    count+=1
                break
            count+=1
        answer.append(count)
        
    return answer

 

• 내 풀이 2
  • 테스트 케이스 추가 : [1, 2, 3, 2, 3, 1] → 결과: [5, 4, 1, 2, 1, 0]
  • 떨어지기까지 버틴 시간을 구하기 
  • 덱(deque) 사용

from collections import deque
def solution(prices):
    queue=deque(prices)
    answer = []
        
    while queue:
        count=0
        q=queue.popleft()
        for a in queue:
            if a<q:
                count+=1
                break
            count+=1
        answer.append(count)
        
    return answer