Lv2. 주식가격

# 문제 설명

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

# 제한 사항

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

# 입출력 예

입출력 예 설명

• 1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
• 2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
• 3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
• 4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
• 5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.

문제 해결 시 고려사항

• 구간 반복의 종료 조건

본 문제를 풀이하기 위해서 사용해야하는 것은 반복문이다. 이 문제는 여러 주식의 가격이 리스트로 주어지고 리스트 내의 각 주식이 떨어지지 않은 기간을 순차적인 리스트로 반환하는 문제이다.

따라서 반복문을 이중으로 사용하여 prices 리스트 내의 각 원소(이하 1번 원소)를 해당 원소의 뒷 원소(이하 2번 원소)들과 비교를 실시한다. 이 비교에서 사용될 조건은 1번 원소가 2번 원소보다 큰 경우에는 반복을 종료한다는 조건이다.

이러한 조건을 이용해 반복문을 실행하면 1번 원소의 가격이 떨어지지 않은 기간을 전체적으로 구할 수 있다.

 # 각 주식을 해당 주식의 뒷 주식들과 자신보다 싼 주식이 나올때까지 검사
     for i in range(len(prices)-1):
         time = 0
         for j in range(i+1, len(prices)):
             time += 1
             if prices[i]>prices[j]:                
                 break
         # 각 주식별로 가격이 떨어지지 않은 기간을 append
         answer.append(time)
     # 가장 마지막 주식은 떨어지지 않음
     answer.append(0)

가장 마지막 주식은 비교할 뒷 원소가 없기 때문에 가격이 떨어질 수 없기 때문에 0을 append하고 answer 리스트를 반환하면 된다.

전체 코드

def solution(prices):
    answer = []
    # 각 주식을 해당 주식의 뒷 주식들과 자신보다 싼 주식이 나올때까지 검사
    for i in range(len(prices)-1):
        time = 0
        for j in range(i+1, len(prices)):
            time += 1
            if prices[i]>prices[j]:                
                break
        # 각 주식별로 가격이 떨어지지 않은 기간을 append
        answer.append(time)
    # 가장 마지막 주식은 떨어지지 않음
    answer.append(0)
    return answer