python3 二分法查找算法及功能扩展

介绍

二分查找顾名思义就是从序列的中间位置查找,都将目标数字与序列的中间位置数字进行对比,如果目标数字等于中间位置数字则返回对应的序列索引,如果目标数字大于中间位置数字,则继续从有侧的序列中利用二分查找,如果目标数字小于中间位置数字,则继续从左侧的序列中利用二分查找,直到查到目标数字为止。二分法查找的效率很高,但是也有其局限性,比如,目标序列必须是有序的序列,查找的目标如果在序列中有多个,只能查找到一个等。
这里介绍了基础的二分查找算法,并且对其进行了简单的扩展,扩展后增加:

  1. 容错功能,当传参不符合要求的时候不会抛出异常
  2. 查全功能,可以查到目标序列中全部的目标数字的位置
  3. 无序序列处理,如果序列为无序序列时,可以选择是做普通查找还是转换为有序序列后做二分查找
  4. 结果格式化,返回结果为一个字典
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、二分查找算法扩展

代码中有详细的注释说明,扩展后增加的内容包括:

  1. 容错功能,当传参不符合要求的时候不会抛出异常
  2. 查全功能,可以查到目标序列中全部的目标数字的位置
  3. 无序序列处理,如果序列为无序序列时,可以选择是做普通查找还是转换为有序序列后做二分查找
  4. 结果格式化,返回结果为一个字典
源码
# 扩展二分法查找数字,返回为一个字典
def halfSearchTargetNum(baseList, targetNum, isSorted=False):
    # 参数说明:
    # baseList   目标序列
    # targetNum   目标数字
    # isSorted  默认为False,如果目标序列不是有序序列时,若为False做普通查找,若为True则排序后做二分查找
    targetIndexList = []
    res = {'查找结果': targetIndexList, '目标序列': baseList, '目标数字': targetNum, '说明': ''}

    # 处理baseList类型错误的情况
    if not isinstance(baseList, (list, tuple)):
        res['说明'] = 'baseList不是列表或元组'
        return res

    # 处理baseList元素类型错误的情况
    for item in baseList:
        if not isinstance(item, (int, float)):
            res['说明'] = 'baseList中有非数字元素'
            return res

    # 处理targetNum类型错误的情况
    if not isinstance(targetNum, (int, float)):
        res['说明'] = '目标数字targetNum不是数字类型'
        return res

    # 处理targetNum不在baseList的情况
    if targetNum not in baseList:
        res['说明'] = '目标数字在目标序列中不存在'
        return res

    res['说明'] = '有序序列,二分查找'

    # 处理baseList不是有序序列的情况
    baseListSorted = sorted(baseList)
    if baseList != baseListSorted:
        # isSorted = True时对目标序列重新排序,再对排序后的序列做二分查找
        if isSorted:
            baseList = baseListSorted
            res['原序列'] = res['目标序列']
            res['目标序列'] = baseList
            res['说明'] = '无序序列转有序序列,二分查找'

        # isSorted = False时不重新排序,做普通查找
        else:
            for i in range(0, len(baseList)):
                if baseList[i] == targetNum:
                    targetIndexList.append(i)
            res['说明'] = '无序序列,普通查找'
            return res

    # 开始二分查找
    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:
            # 二分查找到的结果存入列表中
            targetIndexList.append(midIndex)

            # 在二分查找结果右侧继续查找是否还有目标数字,若有就存进列表中,如果遇到不等于目标数字的立即结束查找(因为是有序的,且从左往右查找)
            for i in range(midIndex + 1, rightIndex + 1):
                if baseList[i] != targetNum:
                    break
                else:
                    targetIndexList.append(i)

            # 在二分查找结果左侧继续查找是否还有目标数字,若有就存进列表中,如果遇到不等于目标数字的立即结束查找(因为是有序的,且从右往左查找)
            while leftIndex < midIndex:
                if baseList[midIndex - 1] != targetNum:
                    break
                else:
                    targetIndexList.append(midIndex - 1)
                midIndex -= 1

            # 将结果列表变为有序列表
            targetIndexList.sort()
            return res
调试
if __name__ == '__main__':
    test1 = [5, 5.1, 5.1, 5.1, 5.11]
    test2 = [-20, -3, 92, 12, -2, 12, 7.6, 8, 56]
    test3 = 99
    test4 = [-20, -3, 92, 12, -2, '12', 7.6, 8, 56]
    print('结果1:',halfSearchTargetNum(test1, 5.1))
    print('结果2:',halfSearchTargetNum(test1, 5.1, isSorted=True))
    print('结果3:',halfSearchTargetNum(test2, 12, isSorted=True))
    print('结果4:',halfSearchTargetNum(test2, 12))
    print('结果5:',halfSearchTargetNum(test2, 100))
    print('结果6:',halfSearchTargetNum(test2, '12'))
    print('结果7:',halfSearchTargetNum(test3, 99))
    print('结果8:',halfSearchTargetNum(test4, 12))
结果
结果1: {'查找结果': [1, 2, 3], '目标序列': [5, 5.1, 5.1, 5.1, 5.11], '目标数字': 5.1, '说明': '有序序列,二分查找'}
结果2: {'查找结果': [1, 2, 3], '目标序列': [5, 5.1, 5.1, 5.1, 5.11], '目标数字': 5.1, '说明': '有序序列,二分查找'}
结果3: {'查找结果': [5, 6], '目标序列': [-20, -3, -2, 7.6, 8, 12, 12, 56, 92], '目标数字': 12, '说明': '无序序列转有序序列,二分查找', '原序列': [-20, -3, 92, 12, -2, 12, 7.6, 8, 56]}
结果4: {'查找结果': [3, 5], '目标序列': [-20, -3, 92, 12, -2, 12, 7.6, 8, 56], '目标数字': 12, '说明': '无序序列,普通查找'}
结果5: {'查找结果': [], '目标序列': [-20, -3, 92, 12, -2, 12, 7.6, 8, 56], '目标数字': 100, '说明': '目标数字在目标序列中不存在'}
结果6: {'查找结果': [], '目标序列': [-20, -3, 92, 12, -2, 12, 7.6, 8, 56], '目标数字': '12', '说明': '目标数字不是数字类型'}
结果7: {'查找结果': [], '目标序列': 99, '目标数字': 99, '说明': '目标序列不是列表或元组'}
结果8: {'查找结果': [], '目标序列': [-20, -3, 92, 12, -2, '12', 7.6, 8, 56], '目标数字': 12, '说明': '目标序列中有非数字元素'}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,099评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,473评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,229评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,570评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,427评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,335评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,737评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,392评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,693评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,730评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,512评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,349评论 3 314
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,750评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,017评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,290评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,706评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,904评论 2 335

推荐阅读更多精彩内容