在 Swift 中使用索引遍历集合的正确且惯用的方法？

Question

我想遍历任意 Swift 集合并获取元素及其索引。

基本上可以替代：

for (idx, el) in collection.enumerate() {
    print("element at \(idx) is \(el)")
}

但这给了我真正的通用索引，而不仅仅是从 0 开始的连续整数。

当然，解决方案将成为接受任何类型集合的通用函数的一部分，否则差异不会很重要。

有没有比下面这样的幼稚循环更好的方法？

var idx = collection.startIndex, endIdx = collection.endIndex
while idx < endIdx {
    let el = collection[idx]
    print("element at \(idx) is \(el)")
    idx = idx.successor()
}

这样的写作似乎很容易出错。我知道我可以将该代码转换为 sn-p，但如果可能的话，我想找到一个更简洁、更惯用的解决方案。

Answer 1

对于任何集合，indices 属性返回有效范围 指数。迭代索引和相应的元素 同时你可以使用zip():

for (idx, el) in zip(collection.indices, collection) {
    print(idx, el)
}

数组切片示例：

let a = ["a", "b", "c", "d", "e", "f"]
let slice = a[2 ..< 5]

for (idx, el) in zip(slice.indices, slice) {
    print("element at \(idx) is \(el)")
}

输出：

2 处的元素是 c 3 处的元素是 d 4 处的元素是 e

您可以为此目的定义自定义扩展方法 （取自How to enumerate a slice using the original indices?）：

// Swift 2:
extension CollectionType {
    func indexEnumerate() -> AnySequence<(index: Index, element: Generator.Element)> {
        return AnySequence(zip(indices, self))
    }
}

// Swift 3:
extension Collection {
    func indexEnumerate() -> AnySequence<(Indices.Iterator.Element, Iterator.Element)> {
        return AnySequence(zip(indices, self))
    }
}

字符视图示例：

let chars = "a???z".characters
for (idx, el) in chars.indexEnumerate() {
    print("element at \(idx) is \(el)")
}

输出：

0 处的元素是 1 处的元素是? 3 的元素是?? 7 处的元素是 z