介绍
二分查找顾名思义就是从序列的中间位置查找,都将目标数字与序列的中间位置数字进行对比,如果目标数字等于中间位置数字则返回对应的序列索引,如果目标数字大于中间位置数字,则继续从有侧的序列中利用二分查找,如果目标数字小于中间位置数字,则继续从左侧的序列中利用二分查找,直到查到目标数字为止。二分法查找的效率很高,但是也有其局限性,比如,目标序列必须是有序的序列,查找的目标如果在序列中有多个,只能查找到一个等。
这里介绍了基础的二分查找算法,并且对其进行了简单的扩展,扩展后增加:
- 容错功能,当传参不符合要求的时候不会抛出异常
- 查全功能,可以查到目标序列中全部的目标数字的位置
- 无序序列处理,如果序列为无序序列时,可以选择是做普通查找还是转换为有序序列后做二分查找
- 结果格式化,返回结果为一个字典
1、基础的二分查找
说明
baseList 必须为有序的数字序列
targetNum 必须为数字
如果查找到目标数字则返回对应的索引,如果没查找到则返回-1
传参类型错误时会抛出异常
源码
# 二分法查找目标数字,baseList必须为有序的数字序列(列表或元组)
def halfSearchTargetNum(baseList, targetNum):
leftIndex = 0
rightIndex = len(baseList) - 1
while leftIndex <= rightIndex:
midIndex = int((leftIndex + rightIndex) / 2)
if baseList[midIndex] < targetNum:
leftIndex = midIndex + 1
elif baseList[midIndex] > targetNum:
rightIndex = midIndex - 1
else:
return midIndex
return -1
调试
if __name__ == '__main__':
test = [-20, -3, 1, 3, 5, 5, 6, 7.6, 8, 56]
print('结果1:', halfSearchTargetNum(test, 5))
print('结果2:', halfSearchTargetNum(test, 7.6))
print('结果3:', halfSearchTargetNum(test, 56))
print('结果4:', halfSearchTargetNum(test, -20))
print('结果5:', halfSearchTargetNum(test, 9))
结果
结果1: 4
结果2: 7
结果3: 9
结果4: 0
结果5: -1
2、二分查找算法扩展
代码中有详细的注释说明,扩展后增加的内容包括:
- 容错功能,当传参不符合要求的时候不会抛出异常
- 查全功能,可以查到目标序列中全部的目标数字的位置
- 无序序列处理,如果序列为无序序列时,可以选择是做普通查找还是转换为有序序列后做二分查找
- 结果格式化,返回结果为一个字典
源码
# 扩展二分法查找数字,返回为一个字典
def halfSearchTargetNum(baseList, targetNum, isSorted=False):<