초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.
| prices | return |
|---|---|
| [1, 2, 3, 2, 3] | [4, 3, 1, 1, 0] |
def solution(prices):
answer = [0] * len(prices) # 각 시점의 가격이 떨어지지 않는 기간을 저장할 리스트를 초기화
for i in range(len(prices)):
for j in range(i+1, len(prices)):
answer[i] += 1 # 가격이 떨어지지 않는 동안 1씩 증가
if prices[i] > prices[j]: # 만약 가격이 떨어진다면, 루프를 빠져나옴
break
return answer