二分查找基于有序数据,通过不断缩小搜索区间实现高效查找,适用于有序数组中找元素、插入位置或边界值,Python的bisect模块可简化操作,处理重复元素时需调整边界以定位首个或末个目标。

在Python中实现二分查找,核心在于利用数据已排序的特性,通过不断将搜索区间减半来高效定位目标元素。这并非什么高深莫测的魔法,更多的是一种对效率的追求,尤其是在处理大规模有序数据集时,它的优势会变得非常明显。
二分查找(Binary Search)的实现通常是迭代式的,因为它直观且避免了递归深度的问题。我们维护两个指针,
low
high
mid
mid
low
high
def binary_search(arr, target):
"""
使用迭代方式实现二分查找。
在有序列表中查找目标元素的位置。
Args:
arr (list): 一个已排序的列表。
target: 要查找的目标元素。
Returns:
int: 如果找到目标元素,返回其索引;否则返回 -1。
"""
low = 0
high = len(arr) - 1
while low <= high:
mid = low + (high - low) // 2 # 避免 (low + high) 溢出,虽然Python整数不常见此问题
if arr[mid] == target:
return mid
elif arr[mid] < target:
low = mid + 1 # 目标在右半部分,更新低位指针
else:
high = mid - 1 # 目标在左半部分,更新高位指针
return -1 # 循环结束仍未找到,说明目标不存在
# 示例用法
sorted_list = [2, 5, 8, 12, 16, 23, 38, 56, 72, 91]
print(f"在 {sorted_list} 中查找 12:索引 {binary_search(sorted_list, 12)}") # 输出:3
print(f"在 {sorted_list} 中查找 38:索引 {binary_search(sorted_list, 38)}") # 输出:6
print(f"在 {sorted_list} 中查找 77:索引 {binary_search(sorted_list, 77)}") # 输出:-1
print(f"在 {sorted_list} 中查找 2:索引 {binary_search(sorted_list, 2)}") # 输出:0
print(f"在 {sorted_list} 中查找 91:索引 {binary_search(sorted_list, 91)}") # 输出:9
print(f"在空列表中查找 5:索引 {binary_search([], 5)}") # 输出:-1
print(f"在单元素列表 [10] 中查找 10:索引 {binary_search([10], 10)}") # 输出:0
print(f"在单元素列表 [10] 中查找 5:索引 {binary_search([10], 5)}") # 输出:-1这个实现算是比较标准和健壮的。它处理了空列表、单元素列表以及目标在边界的情况。
mid = low + (high - low) // 2
low + high
二分查找,顾名思义,其核心原理就是“二分”,也就是分而治之(Divide and Conquer)策略的典型应用。它要求待搜索的数据集必须是有序的,这是其能发挥作用的基石。试想一下,如果数据是乱序的,你随便取个中间值,它的大小并不能告诉你目标在左边还是右边,整个逻辑就崩塌了。正是因为有序,我们才能通过比较中间值与目标的大小,果断地排除掉一半的搜索空间。每一次比较,搜索范围就缩小一半,这种效率提升是指数级的。
立即学习“Python免费学习笔记(深入)”;
它的适用场景非常明确:
bisect
我个人觉得,理解二分查找,不仅仅是记住它的代码,更重要的是领悟“有序”这个前提条件带来的巨大力量。它把一个看似需要线性遍历的问题,硬生生拽到了对数时间复杂度,这在工程实践中是实实在在的性能提升。
bisect
Python标准库中的
bisect
while
mid
bisect
bisect_left(a, x, lo=0, hi=len(a))
a
x
a
x
a
x
x
bisect_right(a, x, lo=0, hi=len(a))
bisect(a, x, lo=0, hi=len(a))
a
x
a
x
a
x
x
insort_left(a, x, lo=0, hi=len(a))
x
a
a
x
x
insort_right(a, x, lo=0, hi=len(a))
insort(a, x, lo=0, hi=len(a))
x
a
a
x
x
我们来看看实际用例:
import bisect
my_list = [1, 3, 3, 6, 8, 12, 15]
# 查找插入点
idx_left = bisect.bisect_left(my_list, 3)
idx_right = bisect.bisect_right(my_list, 3)
idx_new = bisect.bisect_left(my_list, 7) # 查找不存在元素的插入点
print(f"列表中第一个 3 的索引(或其左侧插入点):{idx_left}") # 输出:1
print(f"列表中最后一个 3 的右侧插入点:{idx_right}") # 输出:3
print(f"元素 7 的插入点:{idx_new}") # 输出:4
# 插入元素
bisect.insort_left(my_list, 3)
print(f"插入 3 (insort_left) 后:{my_list}") # 输出:[1, 3, 3, 3, 6, 8, 12, 15]
bisect.insort_right(my_list, 7)
print(f"插入 7 (insort_right) 后:{my_list}") # 输出:[1, 3, 3, 3, 6, 7, 8, 12, 15]bisect
bisect_left
bisect_right
bisect
处理重复元素是二分查找一个比较微妙的地方,因为它直接影响到你最终想找到的是“第一个”重复元素、还是“最后一个”重复元素,或者仅仅是“任意一个”重复元素。我们上面给出的
binary_search
mid
但有时候,需求会更精确:
查找第一个出现的重复元素: 如果目标值存在多个,我们希望找到它们当中索引最小的那一个。这时,当
arr[mid] == target
mid
mid
target
mid
high = mid - 1
def binary_search_first_occurrence(arr, target):
low, high = 0, len(arr) - 1
result = -1
while low <= high:
mid = low + (high - low) // 2
if arr[mid] == target:
result = mid # 找到一个匹配项,记录下来
high = mid - 1 # 继续向左搜索,看有没有更早的匹配
elif arr[mid] < target:
low = mid + 1
else:
high = mid - 1
return result查找最后一个出现的重复元素: 与查找第一个类似,当
arr[mid] == target
mid
low = mid + 1
def binary_search_last_occurrence(arr, target):
low, high = 0, len(arr) - 1
result = -1
while low <= high:
mid = low + (high - low) // 2
if arr[mid] == target:
result = mid # 找到一个匹配项,记录下来
low = mid + 1 # 继续向右搜索,看有没有更晚的匹配
elif arr[mid] < target:
low = mid + 1
else:
high = mid - 1
return result这些变种在面试或者实际工作中处理数据边界时非常常见。比如,你有一个用户行为日志,按时间戳排序,现在想找到某个特定事件在某个时间段内第一次或最后一次发生的位置。这时候,普通的二分查找可能不够用,就需要这些专门处理重复元素的变体。理解这些细微的调整,其实就是对二分查找“收缩边界”逻辑更深层次的把握。同时,你也会发现,
bisect_left
bisect_right
以上就是如何用Python实现二分查找?的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号