초 단위로 기록된 주식가격이 담긴 배열 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)):
cur_price = prices[i]
for j in range(i+1, len(prices)):
if cur_price <= prices[j]:
answer[i] += 1
else:
answer[i] += 1
break
return answer
prices
를 큐에 삽입한 뒤에 pop 하면서 뒤의 가격과 비교한다.