在涉及到一些數(shù)據(jù)結(jié)構(gòu)的經(jīng)典理論和模型 (沒錯(cuò)庆杜,就是鏈表名惩,樹和圖) 時(shí)泌绣,我們往往會(huì)用到嵌套的類型钮追。比如鏈表,在 Swift 中阿迈,我們可以這樣來定義一個(gè)單向鏈表:
class Node<T> {
let value: T?
let next: Node<T>?
init(value: T?, next: Node<T>?) {
self.value = value
self.next = next
}
}
let list = Node(value: 1,
next: Node(value: 2,
next: Node(value: 3,
next: Node(value: 4, next: nil))))
// 單向鏈表:1 -> 2 -> 3 -> 4
看起來還不錯(cuò)元媚,但是這樣的形式在表達(dá)空節(jié)點(diǎn)的時(shí)候并不十分理想。我們不得不借助于 nil 來表達(dá)空節(jié)點(diǎn)苗沧,但是事實(shí)上空節(jié)點(diǎn)和 nil 并不是等價(jià)的刊棕。另外,如果我們想表達(dá)一個(gè)空鏈表的話待逞,要么需要把 list 設(shè)置為 Optional甥角,要么把 Node 里的 value 以及 next 都設(shè)為 nil,這導(dǎo)致描述中存在歧義识樱,我們不得不去做一些人為的約定來表述這樣的情況嗤无,這在算法描述中是十分致命的。
在 Swift 2 中怜庸,我們現(xiàn)在可以使用嵌套的 enum 了 -- 而這在 Swift 1.x 里是編譯器所不允許的当犯。我們用 enum 來重新定義鏈表結(jié)構(gòu)的話,會(huì)是下面這個(gè)樣子:
indirect enum LinkedList<Element: Comparable> {
case Empty
case Node(Element, LinkedList<Element>)
}
let linkedList = LinkedList.Node(1, .Node(2, .Node(3, .Node(4, .Empty))))
// 單項(xiàng)鏈表:1 -> 2 -> 3 -> 4
在 enum 的定義中嵌套自身對(duì)于 class 這樣的引用類型來說沒有任何問題割疾,但是對(duì)于像 struct 或者 enum 這樣的值類型來說嚎卫,普通的做法是不可行的。我們需要在定義前面加上indirect
來提示編譯器不要直接在值類型中直接嵌套宏榕。用 enum 表達(dá)鏈表的好處在于拓诸,我們可以清晰地表示出空節(jié)點(diǎn)這一定義胸懈,這在像 Swift 這樣類型十分嚴(yán)格的語言中是很有幫助的。比如我們可以寥寥數(shù)行就輕易地實(shí)現(xiàn)鏈表節(jié)點(diǎn)的刪除方法恰响,在 enum 中添加:
func linkedListByRemovingElement(element: Element)
-> LinkedList<Element> {
guard case let .Node(value, next) = self else {
return .Empty
}
return value == element ?
next : LinkedList.Node(value, next.linkedListByRemovingElement(element))
}
let result = linkedList.linkedListByRemovingElement(2)
print(result)
// 1 -> 3 -> 4