[프로그래머스][코딩연습/스택/큐]주식가격

programmers.co.kr/learn/courses/30/lessons/42584

코딩테스트 연습 - 주식가격

문제 설명

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

---

from collections import deque
def solution(prices):
    p = deque(prices)
    answer = []
    while(len(p)>0):
        for i in range(len(p)):
            if (p[i] < p[0]) or (i==len(p)-1):
                answer.append(i)
                p.popleft()
                break
    return answer

먼저 취지에 맞게 queue로 풀었다. 하지만  효율성에서 통과하지 못했다.

아이디어는 맨 왼쪽 값과 나머지 값들을 차례대로 비교하고 작거나 같은 수를 발견하면 index를 ans에 append 해 준다.

그리고 맨 왼쪽 값은 pop 해 준다.

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

그래서 그냥 반복문을 이용해서도 풀어봤다. 이렇게하면 효율성 test도 통과한다.

아이디어는 0으로 채워진 answer 리스트에 prices[i]와 같거나 작은 값을 발견할 때 까지 +1 해 주는 것이다.