스택/큐-주식가격

 

 

나의 풀이

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

 

해설

맨 처음 시도한 방법은 if price[i] > price[i+1] 을 활용해서 해당할 경우 카운트에 1을 더해주는 방식으로 접근했었다. 하지만 len(prices)가 5 라고 가정했을 때 for 문 마지막에 prices[5+1]이 되면서 range error가 발생한다. (prices[6] 값은 없기 때문). 따라서 for문을 한 개 더 써서 변수 j를 만들어주고 i와 비교하는 방식을 접근하였다.