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

1. 문제 설명

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

 

2. 제한사항

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

 

3. 입출력 예

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

 

4. solution.py

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

1. 가격이 떨어지지 않는 시간을 기록할 리스트 선언(prices 크기만큼)

2. 현재 주식 가격과 다음 주식 가격을 비교하여 count+=1

3. 현재 주식보다 다음 주식 가격이 낮으면 break 선언