[Python][프로그래머스][Level2] 주식가격

def solution(prices):
    answer=[]
    check=[[0,prices[0]]]
    
    for idx in range(1,len(prices)):
        while check and check[-1][1]>prices[idx]:
            i,a=check.pop()
            answer.append([i,idx-i])
        check.append([idx,prices[idx]])
    
    for idx, ck in check:
        answer.append([idx,len(prices)-1-idx])
    
    return [ck for idx,ck in sorted(answer)]