주식가격 파이썬

출처:https://programmers.co.kr/

 

문제 설명

초 단위로 기록된 주식가격이 담긴 배열 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초간 가격이 떨어지지 않았습니다.

 

 

가격이 떨어지지 않는 그 구간을 리턴하는 문제였다.

 

가장 빠르게 떠올랐던 답안은

큐의 pop(0)을 이용해서 1개씩 뽑아내고 남은 큐에서 값을 검사해 순차적으로 초를 구하는 방식이었다.

 

 

여기서 더 줄일 수 있던게 q가 존재하지 않으면 자동으로  for문도 안돌아가니까

 if not q:
            answer.append(0)
            break

이 부분을 생략해주는게 좋다.

 

그리고 deque를 안써서 효율성에서 떨어졌다.

큐 스택은 반드시 써주는게 좋은듯하다

 

def solution(prices):
    from collections import deque
    q=deque(prices)
    answer = []
    while q:
        time=0
        tmp=q.popleft()

       #if not q:
            #answer.append(0)
            #break

        for i in q:
            time += 1
            if tmp>i:
                break
        answer.append(time)
    return answer