엔지니어 규의 IT 프로그래밍 다이어리

프로그래머스 : 스택/큐 - 주식가격 본문

코딩테스트/python

프로그래머스 : 스택/큐 - 주식가격

엔지니어 규 2023. 1. 22. 15:16
728x90

문제 설명

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

※ 공지 - 2019년 2월 28일 지문이 리뉴얼되었습니다.

 

나의풀이

def solution(prices):
  
    p =[x for x in enumerate(prices)]
    answer = []  
    for i in range(len(prices)):
        if all([prices[i] <= x for x in prices[i:]]):
            answer.append(len(prices) - 1 - i)
        else:
            cnt=0
            for j in prices[i+1:]:
            
                cnt += 1
                if prices[i] > j:
                    answer.append(cnt)
                    break
                
                else:
                    continue
                    
    return answer
 
cs

문제는 잘 맞췄으나.. 시간초과가 났다. 

 

다른풀이

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

간단해보이긴 하지만 이게 왜 효율적인지는 알수 없다.

 

 

728x90
Comments