[프로그래머스]스택,큐/ 주식가격(python)

Effy_ee·2023년 8월 7일
0

코딩테스트

목록 보기
42/118

스택/큐(Lv.02) 주식가격👾
https://school.programmers.co.kr/learn/courses/30/lessons/42584

📖 문제

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

💻답안

당연히 실패

def solution(prices):
    answer = [0]*len(prices)
    for i in range(len(prices)):
        for j in range(i+1,len(prices)):
            if prices[i]<=prices[j]:
                answer[i]+=1
                
    return answer

시간을 줄여야하니까 deque를 사용

from collections import deque

def solution(prices):
    queue = deque(prices)
    answer = []
    
    while queue:
        price = queue.popleft()
        sec = 0
        for q in queue:
            sec += 1
            if price > q:
                break 
        answer.append(sec)        
    return answer

완료😊

0개의 댓글