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

문제 설명

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

 

제한사항

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

 

정답

def solution(prices):
    answer = [i for i in range(len(prices)-1,-1,-1)]
    
    for before in range(len(prices)):
        for after in range(before+1,len(prices)):
            if prices[after] < prices[before]:
                answer[before] = after-before
                break
    
    return answer

---

문제를 이해하는데 오래걸렸다.

 

현재의 시점에서 가격이 떨어질 때까지의 시간을 return 하는 문제였다.

끝까지 떨어지지않는다고 가정하고 prices 길이만큼 내림차순으로 정렬된 answer 리스트를 만들어준 다음

이중 for문을 사용하여 자신보다 큰 가격이 나올 때 까지의 거리를 이용해 answer 값을 바꿔주었다.