将 Swift 序列转换为相邻对

Question

鉴于我在 Swift 中有一个数组，例如 [1,2,3,4]，一个方法 pairs() 会将其转换为元组数组：[(1,2), (2,3), (3,4)]。

这里有更多关于pairs() 行为方式的示例：

• pairs([]) 应该返回 []，因为它没有对。
• pairs([1]) 也应该返回 []，因为它没有对。
• pairs([1,2]) 应该是 [(1,2)]。它只有一对。

我可以为Array 编写代码来执行此操作，但我希望pairs() 作为Sequence 的扩展名可用，以便它返回一对Sequence。这将使其可用于任何序列，并与 map、reduce、filter 等方法兼容。

我该如何创建这样的Sequence？以及如何编写方法以这种方式转换任何Sequence，以便尽可能灵活地使用它？

Answer 1

如果我们定义一个扩展，我们可以使用zip() 和dropFirst() 在Collection 上输入：

extension Collection {
    func pairs() -> AnySequence<(Element, Element)> {
        return AnySequence(zip(self, self.dropFirst()))
    }
}

例子：

let array = [1, 2, 3, 4]
for p in array.pairs() {
    print(p)
}

输出：

(1, 2) (2, 3) (3, 4)

更多示例：

print(Array("abc".pairs()))
// [("a", "b"), ("b", "c")]

print([1, 2, 3, 4, 5].pairs().map(+))
// [3, 5, 7, 9]

print([3, 1, 4, 1, 5, 9, 2].pairs().filter(<))
// [(1, 4), (1, 5), (5, 9)]

---

（不像我在这个答案的第一个版本中写的......）这个 应用于Sequence 时，方法是不安全的，因为它是 不保证一个序列可以被遍历多次 非破坏性的。

这是一个带有自定义迭代器类型的直接实现 这也适用于序列：

struct PairSequence<S: Sequence>: IteratorProtocol, Sequence {
    var it: S.Iterator
    var last: S.Element?

    init(seq: S) {
        it = seq.makeIterator()
        last = it.next()
    }

    mutating func next() -> (S.Element, S.Element)? {
        guard let a = last, let b = it.next() else { return nil }
        last = b
        return (a, b)
    }
}

extension Sequence {
    func pairs() -> PairSequence<Self> {
        return PairSequence(seq: self)
    }
}

例子：

print(Array([1, 2, 3, 4].pairs().pairs()))
// [((1, 2), (2, 3)), ((2, 3), (3, 4))]