Python实现二分查找

二分查找

对于包含$n$个元素的列表,用二分查找最多需要${\log }_{2}n$步,而简单查找最多需要$n$步.
例如:在1-100之间猜数字,假设正确数字是100

• 简单查找:从1开始顺序查找(最长路径),则最多需要100步,才能找到100;
• 二分查找:从50开始查找,每次取一半的位置再查找;第一次50,小了,第二次75,小了,第三次88,小了,第四次94,小了,第五次97小了,第六次99小了,第七次100正确.总共7步.
  $${\log }_{2}100\approx 7(2^6=64,2^7=128)$$
• 仅当列表是有序的时候，二分查找才管用。

# 定义二分查找函数
def binary_search(list, item):
    # 设定下标的上下限,初始化为 0,和 长度-1
    low = 0
    high = len(list) - 1
    while low <= high:
        # 下标需要 int类型 不能是 float类型 ,用整除
        mid = (low + high) // 2
        guess = list[mid]
        # 如果猜对了,返回下标
        if guess == item:
            return mid
        # 如果猜的大了,上限设定为mid-1
        if guess > item:
            high = mid - 1
        # 如果猜的小了,下限设定为mid+1
        else:
            low = mid + 1
    # 不存在,返回None
    return


# 生成1-100的列表
my_list = []
for i in range(1, 101):
    my_list.append(i)

index = binary_search(my_list, 100)  # 99
# index = binary_search(my_list, 1)  # 0
# index = binary_search(my_list, 50)  # 49
# index = binary_search(my_list, -1)  # None
print(index)