Lv.2_뒤에 있는 큰 수 찾기

https://school.programmers.co.kr/learn/courses/30/lessons/154539

프로그래머스

def solution(numbers):
    
    answer = []
    for i in range(len(numbers)-1):
        exist = False
        for j in range(i+1, len(numbers)):
            if numbers[i] < numbers[j]:
                answer.append(numbers[j])
                exist = True
                break
                
        if not exist:
            answer.append(-1)
            
    answer.append(-1)
    
    return answer

초반 코드로 작성 시에는 시간초과가 발생한다.

 

def solution(numbers):
    
    answer = [-1] * len(numbers)
    stack = []
    
    for idx, num in enumerate(numbers):
        
        while stack and numbers[stack[-1]] < num:
            i = stack.pop()
            answer[i] = num
            
        stack.append(idx)
        
    return answer

stack으로 풀어야 시간 초과 없이 통과 가능하다.