뒤에 있는 큰 수 찾기

·2023년 6월 3일

출처: 프로그래머스 코딩 테스트 연습

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

처음엔 아무생각없이 2중 for문으로 돌리니깐 시간복잡도가 무지막지하게 커져서 시간초과가 나옴
힌트보고 스택사용해서 겨우 풀었다.

profile
🐥

0개의 댓글