count 返回一个IndexDistance,它是描述的类型
两个集合索引之间的距离。 IndexDistance 是
必须是SignedInteger,但不必是Int,并且可以
不同于Index。因此无法创建
范围0..<count - 1。
一种解决方案是使用startIndex 和endIndex 而不是0 和count:
extension MutableCollection where Index == Int {
/// Shuffle the elements of `self` in-place.
mutating func shuffle() {
// empty and single-element collections don't shuffle
if count < 2 { return }
for i in startIndex ..< endIndex - 1 {
let j = Int(arc4random_uniform(UInt32(endIndex - i))) + i
if i != j {
swap(&self[i], &self[j])
}
}
}
}
另一个优点是这也适用于数组 slices
(其中第一个元素的索引不一定为零)。
注意根据新的"Swift API Design Guidelines",
shuffle() 是 mutating shuffle 方法的“正确”名称,
和shuffled() 用于返回数组的非变异对应物:
extension Collection {
/// Return a copy of `self` with its elements shuffled
func shuffled() -> [Iterator.Element] {
var list = Array(self)
list.shuffle()
return list
}
}
更新: 一个(更通用的)Swift 3 版本已添加到
同时How do I shuffle an array in Swift?。
对于 Swift 4 (Xcode 9),必须替换对 swap() 的调用
通过调用集合的swapAt() 方法来实现。
也不再需要对 Index 类型的限制:
extension MutableCollection {
/// Shuffle the elements of `self` in-place.
mutating func shuffle() {
for i in indices.dropLast() {
let diff = distance(from: i, to: endIndex)
let j = index(i, offsetBy: numericCast(arc4random_uniform(numericCast(diff))))
swapAt(i, j)
}
}
}
有关swapAt 的更多信息,请参阅SE-0173 Add MutableCollection.swapAt(_:_:)。
从 Swift 4.2(Xcode 10,目前处于测试阶段)开始,实现了
SE-0202 Random Unification,
shuffle() 和 shuffled() 是 Swift 标准库的一部分。