Stack_by_array

Stack_by_array.swift

public struct Stack<T> {
    private var elements = Array<T>()
    public init() {}
    
    public mutating func pop() -> T? {
        return self.elements.popLast()
    }
    
    public mutating func push(element: T) {
        self.elements.append(element)
    }
    
    public func peek() -> T? {
        return self.elements.last
    }
    
    public var isEmpty: Bool {
        return self.elements.isEmpty
    }
    
    public var count: Int {
        return self.elements.count
    }
}
extension Stack: CustomStringConvertible {
    public var description: String {
        return self.elements.description
    }
}
var myStack = Stack<Int>()

for i in 1...5 {
    myStack.push(element: i)
}
print(myStack)