프로그래머스 level 2 - 주식가격 (스택/큐)

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

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

분명 전에 풀었던 문제인데 못풀었다?!?!?!

 

바로 다음 초의 주식이 작아지더라도, 1초동안은 유지한걸로 치는걸 감안해서 작성했더니 오답이 나온다.

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

    return answer

 

그래서 j 인덱스 위치를 탐색하면 우선 cnt를 +1한 후, 주식 가격이 떨어지면 for문을 break하도록 했다.

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[i] > prices[j]:
                break           
    return answer