문제 설명
초 단위로 기록된 주식가격이 담긴 배열 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 만큼 반복한다.
'Algorithm(알고리즘) > 프로그래머스 고득점 Kit' 카테고리의 다른 글
프로그래머스 - 크레인 인형뽑기 게임 2019 카카오 (Stack) Level1파이썬 (0) | 2020.12.20 |
---|---|
프로그래머스 고득점 Kit - 네트워크 (DFS/BFS) Level3 파이썬 (0) | 2020.12.04 |
프로그래머스 고득점 Kit - 타겟 넘버 (DFS/BFS) Level2 파이썬 (0) | 2020.12.04 |
프로그래머스 고득점 Kit - 다리를 지나는 트럭 (스택/큐) Level2 파이썬 (0) | 2020.11.27 |
프로그래머스 고득점 Kit - 프린터 (스택/큐) Level2 파이썬 (0) | 2020.11.24 |