概念
- 棧和數(shù)組差多，但是只有一些有限的功能枪汪。你只能 push 一個元素到棧頂，pop 移除棧頂?shù)脑兀琾eek 查看棧頂?shù)脑?/li>
- 棧的順序是后進(jìn)先出
- 按設(shè)計來說仇箱，棧不允許你查看內(nèi)容。只有 peek 方法允許你查看棧頂
棧的實現(xiàn)
struct Stack {
fileprivate var array: [String] = []
}
Push
mutating func push(_ element: String) {
array.append(element)
}
- 注意這里是將元素添加到數(shù)組末尾东羹，不是插入到開始位置剂桥。因為插入元素是一個 O(n) 復(fù)雜度的操作，這個操作會讓數(shù)組中現(xiàn)存的元素全部移動內(nèi)存位置属提，append 不會
Pop
mutating func pop() -> String? {
return array.popLast()
}
Peek
func peek() -> String? {
return array.last
}
- peek 和 pop 類似权逗，唯一的區(qū)別是去除了 mutating，因為 peek 不急改變 array
CustomStringConvertible
extension Stack: CustomStringConvertible {
var description: String {
let topDivider = "---Stack---\n"
let bottomDivider = "\n-----------\n"
let stackElements = array.reversed().joined(separator: "\n")
return topDivider + stackElements + bottomDivider
}
}
- 這里我們通過遵循 CustomStringConvertible 協(xié)議冤议，這里協(xié)議中只有一個 description 屬性需要實現(xiàn)斟薇，通過更改這個屬性的 get 方法，可以改變 print 輸出的信息
var rwBookStack = Stack()
rwBookStack.push("3D Games by Tutorials")
rwBookStack.push("tvOS Apprentice")
rwBookStack.push("iOS Apprentice")
rwBookStack.push("Swift Apprentice")
print(rwBookStack)
---Stack---
Swift Apprentice
iOS Apprentice
tvOS Apprentice
3D Games by Tutorials
-----------
泛型
以上棧的實現(xiàn)我們只能傳入字符串恕酸，我們通過泛型來讓椏氨酰可以傳入任何類型
struct Stack<Element> {
fileprivate var array: [Element] = []
mutating func push(_ element: Element) {
array.append(element)
}
mutating func pop() -> Element? {
return array.popLast()
}
func peek() -> Element? {
return array.last
}
}
然后修改 description 屬性中的代碼
// 先將元素都轉(zhuǎn)為字符串類型，因為 joined 方法必須元素和傳入的參數(shù)類型一致
let stackElements = array.map { "\($0)" }.reversed().joined(separator: "\n")
