JasonScript

Jason's Research Blog

Development/Algorithm

< 알고리즘 문제 풀이 > 뒤에 있는 큰 수 찾기

jason.bak 2023. 12. 5. 20:30

문제 링크 > 코딩테스트 연습 - 뒤에 있는 큰 수 찾기 | 프로그래머스 스쿨 (programmers.co.kr)

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

솔루션 >

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