Swift 스택(Stack) 구현

스택: 한쪽 끝에서만 자료를 넣거나 뺄 수 있는 자료구조로 선형구조이다 (Last In First Out)

자료를 밀어넣는다: push

넣어둔 자료를 꺼낸다: pop 

선입선출하는 큐와 반대로 스택은 후입선출을 지킨다.

Swift에서는 스택을 구현해서 사용해야 하지만

배열의 append, popLast 를 이용하면 굳이 구현하지않고 배열을 스택처럼 사용할 수 있다.

 

import Foundation

struct Stack<T> {
    var elements: [T] = []
    
    var count: Int {
        return elements.count
    }
    var isEmpty: Bool {
        return elements.isEmpty
    }
    
    mutating func pop() -> T? {
        return elements.popLast()
    }
    mutating func push(_ element: T) {
        element.append(element)
    }
    func pop() -> T? {
        return elements.last
    }
}

pop(): popLast() -> O(1)

push: append() -> O(1)

 

스택에서 꺼낼 원소가 없는경우 nil 을 출력한다.