[프로그래머스/Python][스택/큐] 주식가격

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

 

---

문제

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

 

제한사항

• prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
• prices의 길이는 2 이상 100,000 이하입니다.

 

 

 

풀이

문제 이해하기

• 주식가격이 담긴 배열 prices
• 이 배열에서 각 요소가 가격이 떨어지지 않는 기간 리스트 return

문제 접근 방법

• prices의 배열만큼 반복
• 지금 배열이 가리키는 부분에서 더 작은수 나올 때 까지 넣을 수(cnt) 증가시킴

 

 

소스코드

def solution(prices):
    answer = []

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

    return answer