通过偏移量获取元素到序列中

Question

假设我有这个序列：

let fibSequence = sequence(state: (0, 1), next: { (state) -> Int in
    state = (state.1, state.0 + state.1)
    return state.0
})

我可以做一些事情，比如获取前 5 个元素：

print(Array(fibSequence.prefix(5))) // => [1, 1, 2, 3, 5]

但是我怎样才能just 获得索引 5（第 6 个元素）处的元素？ （或者如果序列在第 5 个元素之前终止，则返回 nil。）我能想到的最好的方法是：

print(fibSequence.dropFirst(5).makeIterator().next() as Any) // => Optional(8)

是否有内置方法来执行此类操作？

Answer 1

我认为您的想法是正确的，目前还没有一种方法可以对序列进行索引，但是使用 Sequence 上的扩展很容易实现：

extension Sequence {
    subscript(index: Int) -> Self.Iterator.Element? {
        return enumerated().first(where: {$0.offset == index})?.element
    }
}

这意味着您可以像这样轻松索引序列：

print(fibSequence[8])

另外，我发现您可以使用此逻辑来索引一个特定元素，从而消除对任何警告的强制转换

let fibSequence = sequence(state: (0, 1), next: { (state) -> Int in
    state = (state.1, state.0 + state.1)
    return state.0
})
print(fibSequence.enumerated().first(where: {$0.offset == 5})?.element)