[코딩테스트] 프로그래머스 뒤에 있는 큰 수 찾기 파이썬(Python)

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

프로그래머스

def solution(numbers):
    stack = []
    result = [-1] * len(numbers)
    for i in range(len(numbers)-1, -1, -1):
        if stack == []:
            stack.append(numbers[i])
            continue
        while stack:
            s = stack.pop()
            if s > numbers[i]:
                result[i] = s
                stack.append(s)
                break 
        stack.append(numbers[i])     
    return result

먼저 빈리스트 stack과 numbers의 길이만큼 -1을 저장하고 있는 리스트 result를 초기화하였습니다.

• result의 인덱스(i)로 접근하여 값을 바꾸어줄 것 입니다.

numbers를 뒤에서부터 읽으면서 반복문을 진행합니다.

 

만약 stack 비어있다면 stack에 numbers[i]를 저장해주고 continue키워드로 인하여 다음 반복으로 넘어갑니다.

 

stack에 값이 들어있다면 while문으로 stack이 빌때까지 반복합니다.

먼저 stack.pop()을 사용하여 stack에 저장되어 있는 마지막값(s)를 추출하고
s와 현재 탐색하고 있는 numbers[i]와 비교하여 s가 더 클 경우에는 result[i]를 s로 변경해주고 다시 stack에 s를 저장합니다. 그리고 break문으로 while문을 탈출합니다.

while문을 탈출하면 stack에 numbers[i]를 저장합니다.

• 여기서 저장해야 뒷 큰수를 확인할 수 있습니다.

result 리스트를 결과로 반환합니다.