[프로그래머스/Python] 뒤에 있는 큰 수 찾기
·
Algorithm/Programmers
def solution(numbers): answer = [-1] stack = [numbers[-1]] for i in range(len(numbers)-2, -1, -1): while stack and stack[-1] numbers[i-1]: stack.append(numbers[i]) answer.reverse() return answer 백준-탑 문제와 유사하다. stack에 큰 숫자들을 쌓는다. 현재 top보다 큰 숫자가 나오면 pop()을 해준다. 스택이 비어있으면, 현재 숫자보다 큰 숫자가 뒤에 없었다는 뜻이므로 -1을 넣어준다. 스택이 남아 있다면, 현재 숫자의 뒤에 있는 큰 수는 top 이다. 그리고 마지막에 현재 값도 넣어주는데, 현재 값이 다음 값보다는 큰 수일 수 있기 때문이다. ..