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

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]

但是我怎样才能只索引为 5 的元素(第 6 个元素)？ (如果序列在第 5 个元素之前终止，则返回 nil。)我能想到的最好的办法是:

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

是否有内置方法来执行这样的操作？

Answer 1

我认为您的想法是正确的，目前没有一种方法可以索引序列，但是通过对序列的扩展可以很容易地实现它:

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

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

print(fibSequence[8])

或者我发现你可以使用这个逻辑来索引一个特定的元素，它删除了对 Any 警告的强制转换

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)