프로그래머스 42584 - 주식가격

https://programmers.co.kr/learn/courses/30/lessons/42584#

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

문제 설명 - 스택/큐

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

prices	return
[1, 2, 3, 2, 3]	[4, 3, 1, 1, 0]

---

문제 풀이

질문 이해하는데 30분 걸렸다... 처음 읽었을 때는 무슨 말인지조차 몰랐다

1. prices 인덱스 0번 1을 기준으로 할 때 다음으로 넘어가면서 내 가격보다 낮아질 때는 없다. 그러니 4가 return

이렇게 해석하고 코드를 적으니 맞다. 

효율성이 O(N^2)시간이 나오는데...이게 맞나?

 

def solution(prices):
    answer = []
    
    for i, v in enumerate(prices):
        count = 0
        for j in range(i + 1, len(prices)):
            if v <= prices[j]:
                count += 1
            else:
                count += 1
                break
        answer.append(count)
    
    return answer