#프로그래머스 파이썬 Lv.2 #주식가격

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

 

문제풀이

1. 이 문제는 prices 리스트 각각의 값들이 몇초 동안 가격이 안떨어지고 있었는지 return 해주면 되는 문제
2. 리스트 prices의 각 원소들을 기준으로 잡고 뒤의 값들과 비교
3. return 은 prices의 리스트의 개수만큼 이므로 for문으로 풀 경우 0으로 초기화
4. 파이썬은 answer = [0] * 개수 하면 개수만큼 0으로 초기화된 리스트가 만들어 진다.
5. 만약 앞에 있는 값이 뒤에 있는 값보다 크면 for문을 나와 개수를 리스트 answer에 넣는다
6. 마지막 리스트 answer을 리턴

 

#solution

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

 

 

 

#다른사람풀이

( https://velog.io/@soo5717/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EC%A3%BC%EC%8B%9D%EA%B0%80%EA%B2%A9-Python ) 큐, 스택을 활용한 풀이방법 참고하기!

 

# 큐 queue를 사용

from collections import deque

def solution(prices):
    queue = deque(prices)
    answer = []
    
    while queue:
        price = queue.popleft()
        sec = 0
        for q in queue:
            sec += 1
            if price > q:
                break 
        answer.append(sec)        
    return answer

# 스택 stack을 사용

# prices = [1, 2, 3, 2, 3, 1] return [5, 4, 1, 2, 1, 0]
def solution(prices):
    length = len(prices)
    
    # answer을 max값으로 초기화  
    answer = [ i for i in range (length - 1, -1, -1)]
    
    # 주식 가격이 떨어질 경우 찾기
    stack = [0]
    for i in range (1, length):
        while stack and prices[stack[-1]] > prices[i]:
            j = stack.pop()
            answer[j] = i - j
        stack.append(i)
    return answer

 

 

 

 

https://programmers.co.kr/learn/courses/30/lessons/42584?language=python3 

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