[프로그래머스/Level2/파이썬3(python3)] 주식가격

[프로그래머스/Level2/파이썬3(python3)] 주식가격

문제

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

제한사항

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

입출력 예

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

입출력 예 설명

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

풀이

def solution(prices):
    answer = [0] * len(prices)
    for i in range(len(prices)-1):
        for j in range(i, len(prices)-1):
            if prices[i] > prices[j]:
                break
            else:
                answer[i] += 1
    return answer

설명

지문을 보고 특정 인덱스의 값을 지정해줬다가 오른쪽으로 진행하면서 더 작은 숫자가 나왔을 때의 인덱스 괴리정도를 각 배열에 저장하면 되겠다! 하는 생각이 들어서 금방 풀었던 문제이다. 하지만 맨 처음 풀었을 땐 enumrate함수를 이용했었는데 정확성 테스트는 다 맞아도 효율성 테스트에서 박살이 나더라. 그래서 range(len(prices[:-1]))이런식으로 했더니 이것도 효율성 테스트에서 박살이 났다. 배열을 자르고 붙이는 과정에서 연산이 좀 더 복잡하게 들어가는 것 같다. 그래서 그냥 인터넷에서 가장 간단한 걸 갖다 붙여서 제출했다. 따흐흑.

그래도 새운 걸 하나 배웠다. enumrate는 느리고, 배열 자르고 붙이는 것도 느리다.