[코테] 주식가격[프로그래머스]

문제:

출처: 프로그래머스 - 주식가격

입출력
prices : [1, 2, 3, 2, 3]
return : [4, 3, 1, 1, 0]

풀이:

1. '시작' ~ 'len(prices)-1' 탐색하며 시작 지점보다 작을 때까지 반복문을 진행한다.
2. 작은 부분이 나오면 반복문의 길이만큼 answer에 append하고 반복문을 빠져나온다.
3. 제일 마지막은 항상 길이가 0이다.

코드:

def solution(prices):
    answer = []
    n = len(prices)
    for i in range(n - 1):
        cnt = 0 # 반복문의 길이 초기화
        for j in range(i, n - 1):
            if prices[i] <= prices[j]:
                cnt += 1
            else:
                break
        answer.append(cnt)
    answer.append(0) # 제일 마지막의 길이는 항상 0
    return answer