[Python] 주식가격

스택 응용 풀이

def solution(prices):
    
    answer = []
    postv = 0
    for i in range(len(prices)):
        
        #계속해서 떨어지지 않는단 가정의 기본 값 삽입
        pos = len(prices)-i-1
        answer.append(pos)
        
        #비교를 위한 값 설정
        if i > 0:
            test = i-1
            testv = prices[i]
            postv = prices[i-1]
        
        while i>0 and postv > testv and test>-1:
            
            #떨어지는 위치까지의 거리 값 변경 및 이전 결과에 대한 break point
            if answer[test]>i-test:
            	answer[test]=i-test
            
            #이전 값과 비교
            test=test-1
            postv=prices[test]
            
    return answer

 

ㅁ문제

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