[Swift] 자료구조 - 스택(Stack)

안녕하세요~

 

오늘은 스택에 대해서 정리하려고 합니다 : )

 

---

 

스택(Stack)

 

위의 그림처럼 특정한 데이터를 쌓고 꺼낼 수 있게 만든 자료구조가 바로 스택입니다.

 

LIFO(Last In First Out)로 마지막에 들어온 데이터가 가장 먼저 나가게 되는 특징을 가지고 있습니다.

 

보통 데이터를 넣는 것을 Push라고 하고 꺼내는 것을 Pop이라고 합니다!

 

✔︎ Swift로 구현해 보기

// 제네릭사용
struct Stack<T> { 
    private var stack: [T] = []
    
    // 스택에 있는 데이터의 개수를 반환
    public var count: Int {
        return stack.count
    }
    
    // 스택이 비어있는지 확인하는 프로퍼티
    public var isEmpty: Bool {
        return stack.isEmpty
    }
    
    // Push 
    public mutating func push(_ element: T) {
        stack.append(element)
    }
    
    // Pop
    public mutating func pop() -> T? {
        return isEmpty ? nil : stack.popLast()
    }
}

 

위와 같이 스택을 만드실 수 있습니다.

구조체로 만들었기에 메서드는 mutating 메서드임을 명시해 주어야 합니다.

 

간단한 예제를 보면

 

var languageStack = Stack<String>()

print(languageStack.isEmpty) // True

languageStack.push("Swift")
languageStack.push("C")

print(languageStack.isEmpty) // false

print(languageStack.count) // 2

print(languageStack.pop()) // "C"

 

 

스택을 생성하고 Swift와 C를 순서대로 push한 이후 pop을 호출하면 

마지막에 넣은 C가 출력되는 것을 확인할 수 있습니다.

 

[ 다음 포스팅은 자료구조 큐입니다 ]

[Swift] 자료구조 - 큐(Queue)

 

 

그럼 이만 👋🏻 👋🏻 👋🏻