[Swift] 프로그래머스154539 뒤에 있는 큰 수 찾기

import Foundation

func solution(_ numbers: [Int]) -> [Int] {
    var stack = [Int]()
    var res = Array(repeating: -1, count: numbers.count)

    for i in 0..<numbers.count {
        while !stack.isEmpty && numbers[stack.last!] < numbers[i] {
            res[stack.removeLast()] = numbers[i]
        }
        stack.append(i)
    }

    return res
}

O(N)의 시간복잡도로 풀어야 시간초과가 나지 않는다.

순회를 한번만 하기 위해 stack을 이용한다.