码迷,mamicode.com
首页 > 编程语言 > 详细

Pythonic版二分查找

时间:2018-04-22 16:05:36      阅读:166      评论:0      收藏:0      [点我收藏+]

标签:查找   函数   一个   lambda   __name__   技术   bubuko   返回   元素   

前提:升序数组,待查元素在数组中。

二分查找:就是一个递归函数c。待查元素a,当前数组中位数b,如果b=a则返回b的索引,b>a则在b左侧的子数组中调用函数c,否则在b右侧子数组中调用函数c。

第一次思考,按着上面的思路编程后的结果:

def binary_search(index, a, value):
    if a[(len(a) - 1) // 2] == value:
        return index + (len(a) - 1) // 2
    elif a[(len(a) - 1) // 2] < value:
        return binary_search(index + (len(a) - 1) // 2 + 1, a[(len(a) - 1) // 2 + 1:], value)
    else:
        return binary_search(index, a[0:(len(a) - 1) // 2 + 1], value)

第二次思考,简化中位数计算逻辑:

def binary_search(index, a, value):
    if a[len(a) // 2] == value:
        return index + len(a) // 2
    elif a[len(a) // 2] < value:
        return binary_search(index + len(a) // 2, a[len(a) // 2:], value)
    else:
        return binary_search(index, a[0:len(a) // 2], value)

第三次思考,去掉return,改为lambda形式:

binary_search = lambda index,a,value: index + len(a) // 2 if a[len(a) // 2] == value else binary_search(index + len(a) // 2, a[len(a) // 2:], value) if a[len(a) // 2] < value else binary_search(index, a[0:len(a) // 2], value)

以上就是二分查找变为“一行代码”版的过程。

运行测试:

if __name__ == __main__:
    a = [1, 2, 33, 43, 52, 66, 88, 99, 111, 120]
    print(f"Target index: {binary_search(0, a, value=33)}")

结果如下:
技术分享图片

Pythonic版二分查找

标签:查找   函数   一个   lambda   __name__   技术   bubuko   返回   元素   

原文地址:https://www.cnblogs.com/LanTianYou/p/8907354.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!