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

* 문제 링크

programmers.co.kr/learn/courses/30/lessons/42584

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

 

* 문제 분류 , 난이도

스택/큐  , Lv2

 

* 문제 요약

초 단위로 기록된 주식가격 배열이 주어질 때, 가격이 떨어지지 않은 기간이 몇초인지 Return 하라.

 

* 풀이 방법

• 2중 for 문을 사용하여 자신보다 뒷 인덱스의 요소들을 검사한다.
• 검사 도중 가격이 하락하면 break
• 가격이 하락하지 않는다면 cnt에 1을 더해준다.
• 최종적으로 cnt (가격이 하락한 요소의 인덱스) - idx(자기 자신의 인덱스) 를 answer에 넣어준다.

 

* 파이썬(Pyhton) 코드

def solution(prices):
    maxIndex = len(prices) - 1
    answer =  [0] * (maxIndex + 1)
    for idx, price in enumerate(prices) :
        cnt = idx
        for i in range(maxIndex - idx) :
            cnt += 1
            if(not price <= prices[cnt]) :
                answer[idx] = cnt - idx
                break
            answer[idx] = cnt - idx
    return answer