[프로그래머스 고득점 kit] 스택/큐 - 주식 가격

문제 설명

초 단위로 기록된 주식가격이 담긴 배열 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초간 가격이 떨어지지 않았습니다.

---

 

 

가격이 떨어지지 않은 초를 return하는 문제

 

 

나의 풀이..

from collections import deque

def solution(prices):
    sec_list = []
    prices = deque(prices)

    while prices:
        now_price = prices.popleft()
        sec = 0

        for check_price in prices:
            if now_price <= check_price:
                sec += 1
            else:
                sec += 1
                break  

        sec_list.append(sec)

    return sec_list

 

 

while와 for로 중첩반복문을 사용했다

상세 설명

from collections import deque

def solution(prices):
    
    #가격이 떨어지지 않은 초(sec)를 넣어줄 list를 생성하고 prices를 데큐로 만들어줌
    sec_list = [] 
    prices = deque(prices)
    
    #prices를 popleft 해주며 반복문 돌렸다
    #왜냐하면 가격의 오른쪽으로만 비교를 해야하기 때문에 pop해주기 위함
    #그리고 뽑아낸 가장 왼쪽것을 now_price에 넣어주고 for문 돌릴 준비 끝!
    while prices:
        now_price = prices.popleft()
        sec = 0
	
    #아까 leftpop해준 값과 나머지 prices를 돌아가며 비교해준다
    #now_price가 비교해줄 check_price보다 작거나 같으면 초를 +1 아니면 반복문을 끝내준다
    #else에도 +1을 해주는 이유는 sec=0으로 시작했고,
    #다음으로 지나가면 1초가 지난다고 문제에 써있어서 +1해줌
        for check_price in prices:
            if now_price <= check_price:
                sec += 1
            else:
                sec += 1
                break  
                
	#for문 밖에서 append를 해주면 끝!
        sec_list.append(sec)

    return sec_list

 

list로 풀수도 있었지만 deque를 사용한 점이 나에겐 큰 의의의의의...