Swift 进阶 - 集合协议

2021-07-26 本文已影响0人 iOS心安

Swift 中集合协议是数组、字典、集合和字符串实现的基础，有一些数据结构和算法的知识，理解这部分内容更容易一些。

Advance Swift

Sequence 和 Iterator

Sequence 就是一串相同类型的值，可以在其上遍历迭代：

protocol Sequence {
    associatedtype Element // 序列中的元素
    associatedtype Iterator: IteratorProtocol // 迭代器协议

    func makeIterator() -> Iterator // 生成迭代器
}

迭代器协议声明就比较简单，next() 方法获取下一个值，不能后退，也不能随机访问，next() 返回 nil 时表明迭代结束，所以也可以一直返回非 nil 的值就永远不结束：

protocol IteratorProtocol {
    associatedtype Element
    mutating func next() -> Element?
}

编译器将 for loop 转换为下面的迭代器代码：

var iterator = someSequence.makeIterator()
while let element = iterator.next() {
    doSomething(with: element)
}

如下的迭代器，输入 “abc”，每次迭代生成 “a”、”ab” 和 “abc”：

struct PrefixIterator: IteratorProtocol {
    let string: String
    var offset: String.Index

    init(string: String) {
        self.string = string
        offset = string.startIndex
    }

    mutating func next() -> Substring? {
        guard offset < string.endIndex else { return nil }
        string.formIndex(after: &offset)
        return string[..<offset]
    }
}

如下的 Sequence 使用了上面的迭代器，这个 Sequence 保存了输入的字符串，makeIterator() 都是根据保存的字符串创建迭代器，然后在这个基础进行一次遍历迭代，可以看出 Sequence 和迭代器的关系，迭代器就只是负责给出下一个值，有可能会改变一些初始值，Sequence 保存了初始值，以便创建迭代器时使用：

struct PrefixSequence: Sequence {
    let string: String

    func makeIterator() -> PrefixIterator {
        return PrefixIterator(string: string)
    }
}

Sequence 通常是只能单次遍历，也有可以多次遍历的情况，其语义并不保证可以多次遍历，尽量实现成只能单次遍历。

Collection

Collection 是可以无破坏多次遍历、有限的序列，还可以通过下标访问。

Collection 协议是基于 Sequence 协议，满足 Collection 协议需要实现内容很多，但是因为有默认实现，最基本的需要实现的内容如下，满足了 Collection 协议后，就可以拥有很多功能，查看 Collection - Apple Developer Documentation：

protocol Collection: Sequence {
    associatedtype Element // 序列中的元素
    associatedtype Index: Comparable // 下标
    var startIndex: Index { get } // 第一个元素的下标
    var endIndex: Index { get } // 最后一个元素的下标 + 1
    func index(after i: Index) -> Index // 当前下标 + 1
    subscript(position: Index) -> Element { get } // 下标获取元素
}

Array Literals 是数组语义，Swift 数组和 Swift 集合中有关于通过如下数组语义创建的代码：

let evenNumbers = [2, 4, 6, 8]
var shoppingList = ["Eggs", "Milk"]

var genres: Set<String> = ["Rock", "Pop", "Jazz"]

通过实现 ExpressibleByArrayLiteral 协议，你也可以拥有：

extension FIFOQueue: ExpressibleByArrayLiteral {
    public init(arrayLiteral elements: Element...) {
        self.init(left: elements.reversed(), right: [])
    }
}

值得注意的 Associated Types：

Iterator - 前面 Sequence 中已经讲过。
Indices - Index 不一定只是通过 Int 实现，满足 Comparable 协议都可以，注意不能有指向 Collection 的引用，否则会有性能问题。查看 Advance Swift - Indices 了解更多。
SubSequence - 也是 Collection，和原 Collection 共享所有元素，只是通过两个下标来限定范围，可以通过 String(substring) 或 Array(arraySlice) 来拷贝元素创建新的 Collection。查看 Advance Swift - Subsequences 了解更多。

集合协议层级

Swift Collection Protocols

BidirectionalCollection

BidirectionalCollection 支持向后遍历和向前遍历。

protocol BidirectionalCollection: Collection {
    func index(before i: Index) -> Index // 当前下标 - 1
}

RandomAccessCollection

RandomAccessCollection 支持常量时间的下标计算。

protocol RandomAccessCollection: BidirectionalCollection {
    func index(_ i: Index, offsetBy distance: Int) -> Index // 根据距离计算下标
    func distance(from start: Index, to end: Index) -> Int // 计算两下标的距离
}

MutableCollection

MutableCollection 支持原址元素修改。

protocol MutableCollection: Collection {
    subscript(position: Index) -> Element { set } // 下标修改元素
}

RangeReplaceableCollection

RangeReplaceableCollection 支持元素新增和元素删除。

protocol RangeReplaceableCollection: Collection {
    init() // 创建空集合
    mutating func replaceSubrange<C>(_ subrange: Range<Index>, with newElements: C) // 替换下标范围的元素
        where C : Collection, Element == C.Element
}

LazySequenceProtocol 和 LazyCollectionProtocol

LazySequenceProtocol 和 LazyCollectionProtocol 支持当需要时才计算。

let filtered = standardIn.lazy.filter {
    $0.split(separator: " ").count > 3
}
for line in filtered {
    print(line)
}

(1..<100).lazy.map { $0 * $0 }.filter { $0 > 10 }.map { "\($0)"}

转载