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

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

프로그래머스

 

- Stack 을 이용하여 현재 수보다 크지 못한 수들의 위치를 push해서 쌓아 올림

- 큰수가 생기면 result의 요소들을 다음으로 큰수로 변경 

 

* 7, 20~23 번 런타임, 실행초과 에러 = For문 중복 + 마지막 요소가 큰 수 일 경우 비효율 적일 수 있음

import Foundation

func solution(_ numbers:[Int]) -> [Int] {
    
    var result = Array(repeating: -1, count: numbers.count)
    
    var stack = [Int]()
    
    for i in 0..<numbers.count{
        
        while !stack.isEmpty && numbers[stack.last!] < numbers[i]{
            result[stack.popLast()!] = numbers[i]
        }
        stack.append(i)
    }
    
    return result
}