스위프트로 구현하는 자료구조 1: 스택(Stack)

스택 (Stack)

스택은 LIFO(Last In First Out)의 특징을 가지는 자료구조입니다. 스택은 자료를 저장할 때, 먼저 넣은 데이터를 가장 마지막에 꺼내게 됩니다. 스위프트의 배열은 스택과 동일한 append, removeLast 메서드를 지원하기 때문에 배열을 한 번 감싸주는 자료구조를 구현하면 쉽게 만들 수 있습니다.

뼈대 만들기

struct Stack<T> {
    var elements: [T] = []
    
    var count : Int {
        return elements.count
    }
    var isEmpty : Bool {
        return elements.isEmpty
    }
}

먼저, 제네릭으로 데이터를 담을 수 있는 배열 하나를 만들어줍니다. 스택에 담긴 데이터의 개수와 스택이 비어있는지 여부는 스위프트에서 제공하는 count 연산 프로퍼티와 isEmpty 연산 프로퍼티를 그대로 사용하면 되겠죠?

 

삽입(push) - O(1)

    mutating func push(_ element: T) {
        elements.append(element)
    }

삽입도 배열의 append 메서드를 그대로 사용합니다.

 

조회(top) - O(1)

    func top() -> T? {
        return elements.last
    }

스택의 조회는 특정한 요소를 조회하는 것이 아니라, 스택의 가장 위에 쌓여있는 요소를 조회할 수 있게 합니다. 따라서 배열의 last 프로퍼티를 통해 마지막 요소를 반환합니다.

 

삭제(pop) - O(1)

    mutating func pop() -> T? {
        return elements.popLast()
    }

삭제도 그대로... 한 가지 팁은 removeLast는 배열이 비어있을 때 사용하면 에러를 발생시키지만, popLast를 사용하면 배열이 비어있을 때는 nil을 반환합니다.

removeLast

popLast

전체 코드

 

스택은 배열 그 자체라서 크게 자세히 작성할 내용이 없네요..! 이렇게 가볍게 시작해서 자료구조 시리즈를 달려보겠습니다! 💪

 

GitHub - jeonyeohun/Data-Structures-In-Swift: 👷🏻‍♂️ STL 왜 없어.. 스위프트로 구현하는 자료구조

자료구조 시리즈의 코드들은 이 레포에서 모두 확인할 수 있어요!