본문 바로가기
Algorithm(알고리즘)/프로그래머스 고득점 Kit

프로그래머스 고득점 Kit - 주식가격 (스택/큐) Level2 파이썬 풀이

by Jun_N 2020. 11. 24.

문제 설명

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

제한사항

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

입출력 예

pricesreturn

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

입출력 예 설명

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

풀이

이 문제는 처음에 문제만 보고 이해하기 좀 힘들었다. 

간단히 이야기 하자면 해당 주식 가격이 끝날때까지 떨어졌는지 체크하는 것인데 떨어졌다면 떨어진 곳까지 몇초나 걸렸는가가 핵심이다.

예를 들어, index0부터 보자면 index 0 (1초 시점)에서는 끝날때까지 주식 가격이 떨어지지 않았으니 4초가 저장되고 2초 시점도 똑같다.

만약에 3초 시점에서는 주식가격이 3이였지만 바로 다음에 2로 떨어졌으니 1초가 걸린다.

해당 로직을 코딩으로 하자면 다음과 같다.  

def solution(prices):
    result = []
    for i in range(len(prices) - 1):
        finish = False
        for j in range(i, len(prices) - 1):
            if prices[i] > prices[j + 1]:
                result.append(j - i + 1)
                finish = True
                break
        if not finish:
            result.append(len(prices) - i - 1)
    result.append(0)
    return result

 

마지막에 result.append(0)을 하는 이유는 마지막 주식 가격은 변경이 없으니 0을 저장해줘야 한다. 

그래서 for문을 전체 주식일 -1 만큼 반복한다.