array.count 和 array[0 ...< index] 会减慢二进制搜索的速度吗？

Question

今天我做了一个工作测试，被要求搜索一个整数数组，这是问题：

本练习的目标是检查数组中是否存在数字。

规格：

项目是按升序排列的整数。

该数组最多可包含 100 万个项目

实现函数existsInArray(_ numbers: [Int], _ k: Int)如果k属于numbers，则返回true，否则函数应返回false。

例子：

let numbers = [-9, 14, 37, 102]
existsInArray(numbers, 102) // returns true
existsInArray(numbers, 36) //returns false

注意：尽量节省 CPU 周期

好的，所以我给出了我的答案，即下面的代码并等待结果

func existsInArray(_ numbers: [Int], _ k: Int) -> Bool {
    if numbers.isEmpty {
        return false
    }
    let numbersHalfIndex: Int = (numbers.count/2)
    if k == numbers[numbersHalfIndex] {
        return true
    } else if k != numbers[0] && numbers.count == 1 {
        return false
    } else if k <= numbers[numbersHalfIndex] {
        let leftHalfNumbersArray = numbers[0 ..< numbersHalfIndex]
        return existsInArray(Array(leftHalfNumbersArray), k)
    } else if k > numbers[numbersHalfIndex] {
        let rightHalfNumbersArray = numbers[numbersHalfIndex ..< numbers.count]
        return existsInArray(Array(rightHalfNumbersArray), k)
    } else {
        return false
    }
}

所以结果是“该解决方案在合理的时间内无法处理一百万个项目”，现在我不知道我做错了什么，因为二分搜索速度快到他妈的。

我唯一的猜测是，可能number.count或numbers[0 ...< numbersHalfIndex]或numbers[numbersHalfIndex ...< number.count]使一切都比预期的要慢。

我是绊倒了还是什么？

编辑：如果有人好奇，我测试了我的代码和 Martin R 代码，看看使用 ArraySlice 在时间方面有多大影响。我使用了从 0 开始按升序排列的 100.000.000 itens 数组。这是我捕获时间的方法：

print("////////// MINE //////////")
var startTime = CFAbsoluteTimeGetCurrent()
print(existsInArray(numbers, 0))
var timeElapsed = CFAbsoluteTimeGetCurrent() - startTime
print("Time elapsed for mine: \(timeElapsed) s.")

print("////////// Martin R //////////")
counter = 0
startTime = CFAbsoluteTimeGetCurrent()
print(existsInArrayOptimal(numbers, 0))
timeElapsed = CFAbsoluteTimeGetCurrent() - startTime
print("Time elapsed for Martin R: \(timeElapsed) s.")

结果如下：

////////// 矿 //////////

真的

我的时间流逝：

1.2008800506591797 秒

///////// 马丁·R//////////

真的

Martin R 经过的时间：0.00012993812561035156 秒。

它大约快 1000 倍！

Answer 1

访问number.count不是问题，因为这是数组的 O(1) 操作。切片numbers[0 ...< numbersHalfIndex]也不是问题。但是Array(leftHalfNumbersArray)从切片创建一个新数组，并复制所有元素。

有两种可能的方法可以避免这种情况：

• 更新数组索引（用于当前搜索范围的下限和上限），而不是创建向下传递递归的数组。
• 向下传递数组切片。切片与原始数组共享元素（只要它们没有发生变异）。

第二种方法的演示：

func existsInArray(_ numbers: ArraySlice<Int>, _ k: Int) -> Bool {
    if numbers.isEmpty {
        return false
    }
    let numbersHalfIndex = numbers.startIndex + numbers.count / 2
    if k == numbers[numbersHalfIndex] {
        return true
    } else if k < numbers[numbersHalfIndex] {
        return existsInArray(numbers[..<numbersHalfIndex], k)
    } else {
        return existsInArray(numbers[(numbersHalfIndex + 1)...], k)
    }
}

请注意，数组切片与原始数组共享它们的索引，因此索引不一定从零开始。这numbers.startIndex就是用于索引计算的原因。

以及一个接受“真实”数组参数的包装函数：

func existsInArray(_ numbers: [Int], _ k: Int) -> Bool {
    return existsInArray(numbers[...], k)
}

正如@Leo 建议的那样，您可以将其实现为一个集合方法，而不是实现两个单独的方法。集合索引不一定是整数，但对于RandomAccessCollection索引计算保证为 O(1)。您还可以将其推广到任意可比较元素而不是整数的集合。

这是一个可能的实现：

extension RandomAccessCollection where Element: Comparable {
    /// Returns a Boolean value indicating whether the collection contains the
    /// given element. It is assumed that the collection elements are sorted
    /// in ascending (non-decreasing) order. 
    ///
    /// - Parameter element: The element to find in the collection.
    /// - Returns: `true` if the element was found in the collection; otherwise,
    ///   `false`.
    ///
    /// - Complexity: O(log(*n*)), where *n* is the size of the collection.
    func binarySearch(for element: Element) -> Bool {
        if isEmpty {
            return false
        }
        let midIndex = index(startIndex, offsetBy: count / 2)
        if element == self[midIndex] {
            return true
        } else if element < self[midIndex] {
            return self[..<midIndex].binarySearch(for: element)
        } else {
            return self[index(after: midIndex)...].binarySearch(for: element)
        }
    }
}

用法：

let numbers = [-9, 14, 37, 102]
print(numbers.binarySearch(for: 102)) // true
print(numbers.binarySearch(for: 36))  // false

或者，一种更新搜索范围索引的非递归方法：

extension RandomAccessCollection where Element: Comparable {
    func binarySearch(for element: Element) -> Bool {
        var lo = startIndex
        var hi = endIndex

        while lo < hi {
            let mid = index(lo, offsetBy: distance(from: lo, to: hi) / 2)
            if element == self[mid] {
                return true
            } else if element < self[mid] {
                hi = mid
            } else {
                lo = index(after: mid)
            }
        }
        return false
    }
}