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

프로그래머스

---

성공 풀이

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

풀이 키워드

스택

예시)
입력값: [9, 1, 5, 3, 6, 2]

해당 for문을 끝낼 때 마다 스택과 result배열의 상태

for i in (0..<numbers.count).reversed() {

스택의 상태:            [2]
result 배열의 상태: [0,0,0,0,0,-1]

스택의 상태:            [6]
result 배열의 상태: [0,0,0,0,-1,-1]

스택의 상태:            [6,3]
result 배열의 상태: [0,0,0,6,-1,-1]

스택의 상태:            [6,5]
result 배열의 상태: [0,0,6,6,-1,-1]

스택의 상태:            [6,5,1]
result 배열의 상태: [0,5,6,6,-1,-1]

스택의 상태:            []
result 배열의 상태: [-1,5,6,6,-1,-1]

출력: [-1,5,6,6,-1,-1]

---