프로그래머스 [Lv. 2] 뒤에 있는 큰 수 찾기 {언어 : Python} [다시 풀어 보기]

문제 링크

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

프로그래머스

정답 코드

def solution(numbers):
    n = len(numbers)
    stack = []
    answer = [-1 for _ in range(n)]
    for i in range(n):
        while stack:
            j = stack.pop()
            if numbers[j] >= numbers[i]:
                stack.append(j)
                break;
            else:
                answer[j] = numbers[i]
        stack.append(i)
            
    return answer

풀이 방법

1. answer를 -1로 채운다.
2. stack이 존재하면
   1. 위치값 j를 빼서 현재 위치값 i와 값을 비교한다.
      1. stack에 있는 값이 크거나 같을 경우 아직 뒤에 큰 수를 찾지 못했다는 말이므로 다시 넣고 끝낸다.
      2. 하지만 현재 값이 stack의 값보다 크다면 현재 값이 뒤에 있는 큰 수가 되므로 answer에 값을 넣어준다.
   2. 현재 위치값도 넣어준다.
3. answer를 출력하면 끝.

느낀점

• 이런식으로 stack을 활용하면 되는데 아직 스택 활용 역량이 부족해보인다.