将 Swift 序列切分为头部和尾部

栏目: Swift · 发布时间: 4年前

内容简介：作者：Ole Begemann，译者：函数式编程语言的一个常用范式是把一个列表切分为头部（第一个元素）和尾部（其余元素）。在 Haskell 中，

作者：Ole Begemann，原文链接，原文日期：2018-11-29

译者： WAMaker ；校对： numbbbbb ， BigNerdCoding ；定稿： Forelax

函数式编程语言的一个常用范式是把一个列表切分为头部（第一个元素）和尾部（其余元素）。在 Haskell 中， x:xs 会匹配非空列表，将头部绑定给变量 x，尾部绑定给 xs。

Swift 不是一门函数式编程语言。既没有内置的 List 类型，也没有集合的特定匹配语法。

集合（Collections）

尽管如此，将 Sequence 或 Collection 切分成头部和尾部偶尔很有用。对于集合来说这很容易：

extension Collection {
    var headAndTail: (head: Element, tail: SubSequence)? {
        guard let head = first else { return nil }
        return (head, dropFirst())
    }
}

if let (firstLetter, remainder) = "Hello".headAndTail {
    // firstLetter: Character == "H"
    // remainder: Substring == "ello"
}

序列（Sequence）

对于序列来说却很困难，因为它们可以是单向（single-pass）的：一些序列只能被迭代一次，迭代器会消耗其中的元素。以网络流为例，一旦你从缓冲区里读取了一个字节，操作系统便将它抛弃了。你无法让它重新来一遍。

一个可能的解决方案是创建一个迭代器读取第一个元素，并把当前的迭代器状态包裹进一个新的 AnySequence 实例：

extension Sequence {
var headAndTail: (head: Element, tail: AnySequence<Element>)? {
    var iterator = makeIterator()
    guard let head = iterator.next() else { return nil }
    let tail = AnySequence { iterator }
    return (head, tail)
}
}

以上代码能够实现功能，但不是一个好的通用解决方案，尤其是对满足 Collection 的类型而言。将尾部包进 AnySequence 会是一个性能杀手，也不可以使用合适的集合类型 SubSequence 。

为了保护集合的 SubSequence 类型，最好给 Collection 和 Sequence 分别写扩展。（我们也将会看到，这是 Swift 5 所推崇的方案，这点会在后面谈到。）

保护 SubSequence 类型

我没有找到一个通用的方案，能够让尾部的 SubSequence 类型完好，也同时能让单向序列正常工作。很感谢 Dennis Vennink 能够找出一个解决方案并分享给我。下面是他的代码（我略微对格式进行了修改）：

extension Sequence {
var headAndTail: (head: Element, tail: SubSequence)? {
    var first: Element? = nil
    let tail = drop(while: { element in
        if first == nil {
            first = element
            return true
        } else {
            return false
        }
    })
    guard let head = first else {
        return nil
    }
    return (head, tail)
}
}

Dennis 的窍门是调用 Sequence.drop(while:) ，为尾部保留了 SubSequence 类型，同时在 drop(while:) 内部“捕获”了第一个元素。干得漂亮！

Swift 5

上面的代码使用 Swift 4.2。在 Swift 5 中由于序列不再会有关联 SubSequence 类型，只存在于集合中（ Swift Evolution proposal SE-0234 ），以上代码会崩溃。

这个改变有很多优势，但同样意味着不可能有一种通用的方法能够让 SubSequence 同时对 Sequence 和 Collection 有效。

相对的，我们把那个简单的解决方案添加给 Collection ：

extension Collection {
var headAndTail: (head: Element, tail: SubSequence)? {
    guard let head = first else { return nil }
    return (head, dropFirst())
}
}

如果我们需要让 Sequence 拥有同样的功能，就需要添加一个独立的扩展，使用新的 DropWhileSequence 作为返回类型的尾部：

extension Sequence {
var headAndTail: (head: Element, tail: DropWhileSequence<Self>)? {
    var first: Element? = nil
    let tail = drop(while: { element in
        if first == nil {
            first = element
            return true
        } else {
            return false
        }
    })
    guard let head = first else {
        return nil
    }
    return (head, tail)
}
}

（实现和之前的代码一样，仅仅改变了返回的类型。）

为集合添加一种模式匹配结构作为一个可行的特性已经在论坛多次被提及。有了它，你可以像下面这样将一个有序集合解构成头部和尾部：

let numbers = 1...10
let [head, tail...] = numbers
// head == 1
// tail == 2...10

在 switch 表达式中会很有用。

很遗憾我们被误导性的名字 Sequence 给束缚住了。要将 Collection.SubSequence 重命名成更合适的 Slice 会造成严重的代码破坏。

本文由 SwiftGG 翻译组翻译，已经获得作者翻译授权，最新文章请访问http://swift.gg。

以上就是本文的全部内容，希望本文的内容对大家的学习或者工作能带来一定的帮助，也希望大家多多支持码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。

码农书籍

Linux内核设计的艺术

新设计团队 / 机械工业出版社华章公司 / 2011-6-20 / 79.00元

关于Linux内核的书已经不计其数，但这本书却是独树一帜的，它的内容代表着Linux内核研究成果的世界顶尖级水平，它在世界范围内首次提出并阐述了操作系统设计的核心指导思想——主奴机制，这是所有操作系统研究者的一笔宝贵财富。本书可能也代表着同类图书的顶尖水平，是一本真正能引导我们较为容易地、极为透彻地理解Linux内核的经典之作，也可能是当前唯一能从本质上指引我们去设计和开发拥有自主知识产权的操作系......一起来看看《Linux内核设计的艺术》这本书的介绍吧!

码农工具