[코테 연습] 주식 가격 Python

문제 설명

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

제한사항

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

---

 

O(n^2)가 안되게 할 방법은 없을지 ㄴ참을 고민해봤지만,, 나의 실력이 부족한 탓인지 전혀 떠오르지 않는다.

그래도 통과는 되었다 ^>^!

def solution(prices):
    n = len(prices)
    for i in range(n-1):
        for j in range(i+1, n):
            if prices[i] > prices[j]:
                prices[i] = j-i
                break
            if j == n-1:
                prices[i] = j-i

    prices[-1] = 0

    return prices