Swift 从 ArraySlices 返回 Array

Question

我正在使用一个排序函数，该函数接受一个已经按降序排序的 Int 数组，并将一个新的 Int 放置在正确的位置。 （即，如果我的排序数组是 [10, 7, 2] 并且新的 int 是 5，则该函数将返回 [10, 7, 5, 2]）。执行此操作的函数一旦找到新 Int 的正确位置，就会将原始数组切片为新 Int 位置之前和之后的项目，然后将切片与新 Int 组合。

我遇到的问题是这不会给我一个数组，而是一个数组切片。

代码：

func addToSorted(sorted: [Int], new: Int) ->  [Int] {
    if sorted.count == 0 {
        return [new]
    } else {
        for index in 0..<sorted.count {
            let item = sorted[index]
            if new > item {
                return sorted[..<index] + [new] + sorted[index...]
            }
        }
    }
}

let result = addToSorted(sorted: [10, 7, 2], new: 5)
print(result) // expected [10, 7, 5, 2]

Answer 1

这是一个更通用（和高效）的替代方案，使用二分搜索

extension RandomAccessCollection where Element : Comparable {
    func descendingInsertionIndex(of value: Element) -> Index {
        var slice : SubSequence = self[...]
        
        while !slice.isEmpty {
            let middle = slice.index(slice.startIndex, offsetBy: slice.count / 2)
            if value > slice[middle] {
                slice = slice[..<middle]
            } else {
                slice = slice[index(after: middle)...]
            }
        }
        return slice.endIndex
    }
}

并使用它

var array = [10, 7, 5, 2]
let index = array.descendingInsertionIndex(of: 4)
array.insert(4, at: index)
print(array) // [10, 7, 5, 4, 2]

对于升序，将if value > slice[middle] 替换为if value < slice[middle]，将return slice.endIndex 替换为return slice.startIndex

Answer 2

如果你使用Swift Algorithms，这个插入是单行的：

var arr = [10, 7, 2]
arr.insert(5, at: arr.partitioningIndex {$0 < 5})
print (arr) // [10, 7, 5, 2]

这是非常有效的 - O(log n) - 因为您的数组已经分区（排序），因此它使用二进制搜索。

Answer 3

您必须将切片提升为数组：

return Array(sorted[..<index]) + [new] + Array(sorted[index...])

其他几点：

1. 您应该养成使用sorted.isEmpty 而不是sorted.count == 0 的习惯，对于一些不存储其计数的集合（例如惰性集合甚至字符串（IIRC））来说，它要快得多。

2. 更好的方法是只使用Array.insert(_:at:):

   var sorted = sorted // Make a local mutable copy
   sorted.insert(new, at: index)
   

3. 顺便说一句，在您的 for 循环之后，您需要在数组末尾插入（这也消除了检查空大小写的需要）：

   return sorted + [new]
   

   由于即使sorted 为空也有效，因此您可以删除该特殊情况。

4. 由于您知道您的数据结构已经排序，您可以使用二分查找而不是线性查找来更快地找到插入索引。