算法通关手册：06 数组二分查找

06 数组二分查找.png

• 本文首发于：「算法通关手册」
• 文中代码地址（欢迎 「Star ★ 」 和 「Fork」）：https://github.com/itcharge/LeetCode-Py

1. 算法介绍

「二分查找算法（Binary Search Algorithm）」，也叫做 「折半查找算法」、「对数查找算法」。是一种在有序数组中查找某一特定元素的搜索算法。

基本算法思想：先确定待查找元素所在的区间范围，在逐步缩小范围，直到找到元素或找不到该元素为止。

二分查找算法的过程如下所示：

• 每次查找时从数组的中间元素开始，如果中间元素正好是要查找的元素，则搜索过程结束；
• 如果某一特定元素大于或者小于中间元素，则在数组大于或小于中间元素的那一半中查找，而且跟开始一样从中间元素开始比较。
• 如果在某一步骤数组为空，则代表找不到。

举个例子来说，给定一个有序数组 [0, 1, 2, 3, 4, 5, 6, 7, 8]。如果我们希望查找 5 是否在这个数组中。

• 第一次区间为整个数组 [0, 1, 2, 3, 4, 5, 6, 7, 8]，中位数是 4，因为 4 小于 5，所以如果 5 存在在这个数组中，那么 5 一定在 4 右边的这一半区间中。于是我们的查找范围变成了 [4, 5, 6, 7, 8]。
• 第二次区间为 [4, 5, 6, 7, 8]，中位数是 6，因为 5 小于 6，所以如果 5 存在在这个数组中，那么 5 一定在 6 左边的这一半区间中。于是我们的查找范围变成了 [4, 5, 6]。
• 第三次区间为 [4, 5, 6]，中位数是 5，正好是我们需要查找的数字。

于是我们发现，对于一个长度为 9 的有序数组，我们只进行了 3 次查找就找到了我们需要查找的数字。而如果是依次遍历数组，则最坏情况下，我们需要查找 9 次。

整个查找过程的示意图如下所示：

image

2. 算法思想

二分查找算法是经典的 「减而治之」 的思想。

这里的 「减」 是减少问题规模的意思，「治」 是解决问题的意思。「减」 和 「治」 结合起来的意思就是 「排除法解决问题」。即：每一次查找，排除掉一定不存在目标元素的区间，在剩下可能存在目标元素的区间中继续查找。每一次通过一些条件判断，将待搜索的区间逐渐缩小，以达到「减少问题规模」的目的。而于问题的规模是有限的，经过有限次的查找，最终会查找到目标元素或者查找失败。

3. 简单二分查找

下面通过一个简单的例子来讲解下二分查找的思路和代码。

• 题目链接：704. 二分查找

3.1 题目大意

给定一个升序的数组 nums，和一个目标值 target，返回 target 在数组中的位置，如果找不到，则返回 -1。

3.2 解题思路

利用二分查找算法进行解题。

设定左右节点为数组两端，即 left = 0，right = len(nums) - 1，代表待查找区间为 [left, right]（左闭右闭）。

取两个节点中心位置 mid，先比较中心位置值 nums[mid] 与目标值 target 的大小。

• 如果中心位置值 nums[mid] 与目标值 target 相等，则返回中心位置。
• 如果中心位置值 nums[mid] 小于目标值 target，则将左节点设置为 mid + 1，然后继续在右区间 [mid + 1, right] 搜索。
• 如果中心位置值 nums[mid] 大于目标值 target，则将右节点设置为 mid - 1，然后继续在左区间 [left, mid - 1] 搜索。

2.3 解题代码

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums) - 1
        # 在区间 [left, right] 内查找 target
        while left <= right:
            # 取区间中间节点
            mid = (left + right) // 2
            # 如果找到目标值，则直接返回中心位置
            if nums[mid] == target:
                return mid
            # 如果 nums[mid] 小于目标值，则在 [mid + 1, right] 中继续搜索
            elif nums[mid] < target:
                left = mid + 1
            # 如果 nums[mid] 大于目标值，则在 [left, mid - 1] 中继续搜索
            else:
                right = mid - 1
        # 未搜索到元素，返回 -1
        return -1

4. 二分查找细节

从上面的例子中我们了解了二分查找的思路和具体代码。但是真正在解决二分查找题目的时候还是需要考虑很多细节的。比如说以下几个问题：

1. 区间的开闭问题：区间应该是左闭右闭，还是左闭右开？
2. mid 的取值问题：mid = (left + right) // 2，还是 mid = (left + right + 1) // 2？
3. 出界条件的判断：left <= right，还是 left < right？
4. 搜索区间范围的选择：left = mid + 1、right = mid - 1、 left = mid、right = mid 应该怎么写？

下面一一讲解。

4.1 区间的开闭问题

区间的左闭右闭、左闭右开指的是初始待查找区间的范围。

• 左闭右闭：初始化赋值时，left = 0，right = len(nums) - 1，left 为数组第一个元素位置，right 为数组最后一个元素位置，从而区间 [left, right] 左右边界上的点都能取到。
• 左闭右开：初始化赋值时，left = 0，right = len(nums)，left 为数组第一个元素位置，right 为数组最后一个元素的下一个位置，从而区间 [left, right) 左边界点能取到，而右边界上的点不能取到。

关于左闭右闭、左闭右开，其实在网上都有对应的代码和解法。但是相对来说，左闭右开这种写法在解决问题的过程中，需要考虑的情况更加复杂，所以建议 全部使用「左闭右闭」区间。

4.2 mid 的取值问题

在二分查找的实际问题中，最常见的 mid 取值就是 mid = (left + right) // 2 或者 mid = left + (right - left) // 2。前者是最常见写法，后者是为了防止整型溢出。式子中 // 2 就代表的含义是中间数「向下取整」。当待查找区间中有偶数个元素个数时，则位于最中间的数为 2 个，这时候使用上面式子只能取到中间靠左边那个数，而取不到中间靠右边的那个数。那么，右边的那个数到底能取吗？

其实，右边的数也是可以取的，令 mid = (left + right + 1) // 2，或者 mid = left + (right - left + 1) // 2。这样如果待查找区间的元素为偶数个，就能取到中间靠右边的那个数了，把这个式子代入到 704. 二分查找 中试一试，发现也是能通过题目评测的。

这是因为二分查找的思路是根据每次选择中间位置上的数值来决定下一次在哪个区间查找元素。每一次选择的元素位置可以是中间位置，但并不是一定非得是区间中间位置元素，靠左一些、靠右一些、甚至区间三分之一、五分之一处等等，都是可以的。比如说 mid = left + (right - left + 1) * 1 // 5 也是可以的。

但一般来说，取中间位置元素在平均意义下所达到的效果最好。同时这样写最简单。而对于 mid 值是向下取整还是向上取整，大多数时候是选择不加 1。但有些写法中，是需要考虑加 1 的，这个后面会说这种写法。

4.3 出界条件的判断

我们经常看到二分查找算法的写法中，while 语句出界判断的语句有left <= right 和 left < right 两种写法。那我们究竟应该在什么情况用什么写法呢？

• 如果判断语句为 left <= right，且查找的元素不存在，则 while 判断语句出界条件是 left == right + 1，写成区间形式就是 [right + 1, right]，此时待查找区间为空，待查找区间中没有元素存在，所以此时终止循环可以直接返回 -1 是正确的。
  • 比如说区间 [3, 2]，不可能存在一个元素既大于等于 3 又小于等于 2，此时直接终止循环，返回 -1 即可。
• 如果判断语句为left < right，且查找的元素不存在，则 while 判断语句出界条件是 left == right，写成区间形式就是 [right, right]。此时区间不为空，待查找区间还有一个元素存在，并不能确定查找的元素不在这个区间中，此时终止循环返回 -1 是错误的。
  • 比如说区间 [2, 2]，元素 2 就属于这个区间，此时终止循环，返回 -1 就漏掉了这个元素。

但是如果我们还是想要使用 left < right 的话，怎么办？

可以在返回的时候需要增加一层判断，判断 left 所指向位置是否等于目标元素，如果是的话就返回 left，如果不是的话返回 -1。即：

# ...
    while left < right:
        # ...
    return left if nums[left] == target else -1

此外，用 left < right 的话还有一个好处，就是退出循环的时候 left == right 成立，就不用判断应该返回 left 还是 right 了。

4.4 搜索区间范围的选择

在进行区间范围选择的时候，有时候是 left = mid + 1、right = mid - 1，还有的时候是 left = mid + 1、right = mid，还有的时候是 left = mid、right = mid - 1。那么我们到底应该如何确定搜索区间范围呢？

这是二分查找的一个难点，写错了很容易造成死循环，或者得不到正确结果。

这其实跟二分查找算法的两种不同思路有关。

• 思路 1：「直接找」—— 在循环体中找到元素后直接返回结果。
• 思路 2：「排除法」—— 在循环体中排除目标元素一定不存在区间。

接下来我们具体讲解下这两种思路。

5. 二分查找两种思路

5.1 思路 1：「直接找」

第 1 种思路比较简单，一旦我们在循环体中找到元素就直接返回结果。其实我们在上边 「3. 简单二分查找 - 704. 二分查找」 中就已经用过了。这里再看一下思路和代码：

思路：

• 取两个节点中心位置 mid，先看中心位置值 nums[mid]。

  • 如果中心位置值 nums[mid] 与目标值 target 相等，则 直接返回 这个中心位置元素的下标。
  • 如果中心位置值 nums[mid] 小于目标值 target，则将左节点设置为 mid + 1，然后继续在右区间 [mid + 1, right] 搜索。
  • 如果中心位置值 nums[mid] 大于目标值 target，则将右节点设置为 mid - 1，然后继续在左区间 [left, mid - 1] 搜索。

代码：

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums) - 1
        # 在区间 [left, right] 内查找 target
        while left <= right:
            # 取区间中间节点
            mid = left + (right - left) // 2
            # 如果找到目标值，则直接范围中心位置
            if nums[mid] == target:
                return mid
            # 如果 nums[mid] 小于目标值，则在 [mid + 1, right] 中继续搜索
            elif nums[mid] < target:
                left = mid + 1
            # 如果 nums[mid] 大于目标值，则在 [left, mid - 1] 中继续搜索
            else:
                right = mid - 1
        # 未搜索到元素，返回 -1
        return -1

细节：

• 这种思路是在一旦循环体中找到元素就直接返回。
• 循环可以继续的条件是 left <= right。
• 如果一旦退出循环，则说明这个区间内一定不存在目标元素。

5.2 思路 2：「排除法」

第 2 种思路在循环体中排除目标元素一定不存在区间。

思路：

• 取两个节点中心位置 mid，根据判断条件先将目标元素一定不存在的区间排除。
• 然后在剩余区间继续查找元素，继续根据条件排除不存在的区间。
• 直到区间中只剩下最后一个元素，然后再判断这个元素是否是目标元素。

根据第二种排除法的思路，我们可以写出来两种代码。

第一种代码：

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums) - 1
        # 在区间 [left, right] 内查找 target
        while left < right:
            # 取区间中间节点
            mid = left + (right - left) // 2
            # nums[mid] 小于目标值，排除掉不可能区间 [left, mid]，在 [mid + 1, right] 中继续搜索
            if nums[mid] < target:
                left = mid + 1 
            # nums[mid] 大于等于目标值，目标元素可能在 [left, mid] 中，在 [left, mid] 中继续搜索
            else:
                right = mid
        # 判断区间剩余元素是否为目标元素，不是则返回 -1
        return left if nums[left] == target else -1

第二种代码：

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums) - 1
        # 在区间 [left, right] 内查找 target
        while left < right:
            # 取区间中间节点
            mid = left + (right - left + 1) // 2
            # nums[mid] 大于目标值，排除掉不可能区间 [mid, right]，在 [left, mid - 1] 中继续搜索
            if nums[mid] > target:
                right = mid - 1 
            # nums[mid] 小于等于目标值，目标元素可能在 [mid, right] 中，在 [mid, right] 中继续搜索
            else:
                left = mid
        # 判断区间剩余元素是否为目标元素，不是则返回 -1
        return left if nums[left] == target else -1

细节：

• 判断语句是 left < right。这样在退出循环时，一定有left == right 成立，就不用判断应该返回 left 还是 right 了。同时方便定位查找元素的下标。但是一定要注意最后要对区间剩余的元素进行一次判断。
• 在循环体中，优先考虑 nums[mid] 在什么情况下一定不是目标元素，排除掉不可能区间，然后再从剩余区间中确定下一次查找区间的范围。
• 在考虑 nums[mid] 在什么情况下一定不是目标元素之后，它的对立面（即 else 部分）一般就不需要再考虑区间范围了，直接取上一个区间的反面区间。如果上一个区间是 [mid + 1, right]，那么相反面就是 [left, mid]。如果上一个区间是 [left, mid - 1]，那么相反面就是 [mid, right]。
• 区分被分为两部分： [left, mid - 1] 与 [mid, right] 时，mid 取值要向上取整。即 mid = left + (right - left + 1) // 2。因为如果当区间中只剩下两个元素时（此时 right = left + 1），一旦进入 left = mid 分支，区间就不会再缩小了，下一次循环的查找区间还是 [left, right]，就陷入了死循环。
• 关于边界设置可以记忆为：只要看到 left = mid 就向上取整。或者记为：
  • left = mid + 1 和 right = mid 和 mid = left + (right - left) // 2 一定是配对出现的。
  • right = mid - 1 和 left = mid 和 mid = left + (right - left + 1) // 2 一定是配对出现的。

5.3 两种思路适用范围

• 二分查找的思路 1：因为判断语句是 left <= right，有时候要考虑返回是 left 还是 right。循环体内有 3 个分支，并且一定有一个分支用于退出循环或者直接返回。这种思路适合解决简单题目。即要查找的元素性质简单，数组中都是非重复元素，且 ==、>、< 的情况非常好写的时候。
• 二分查找的思路 2：更加符合二分查找算法的减治思想。每次排除目标元素一定不存在的区间，达到减少问题规模的效果。然后在可能存在的区间内继续查找目标元素。这种思路适合解决复杂题目。比如查找一个数组里可能不存在的元素，找边界问题，可以使用这种思路。

参考资料

• 【博文】Learning-Algorithms-with-Leetcode - 第 3.1 节 二分查找算法
• 【博文】二分法的细节加细节 你真的应该搞懂！！！_小马的博客
• 【课程】零起步学算法 - LeetBook - 二分查找的基本思想：减而治之
• 【题解】二分查找算法细节详解，顺便写了首诗 - LeetCode