[프로그래머스-Lv2] 주식가격 / Python

코딩테스트 연습 - 주식가격

import sys

input = sys.stdin.readline

prices = list(map(int, input().split()))


def solution(prices):
    answer = []
    for i in range(len(prices)):
        cnt = 0
        for j in range(i + 1, len(prices)):
            cnt += 1
            if prices[i] > prices[j]:
                break
        answer.append(cnt)
    return answer


print(solution(prices))

 

해설

현재 값과 이후 값들을 비교해서 현재 값이 더 큰 시점의 시간을 answer에 저장한다.