二分查找两种实现,附详细注释

Python与算法社区

共 626字,需浏览 2分钟

 · 2020-10-30

二分查找问题描述

给定一个  个元素有序的(升序)整型数组  和一个目标值   ,写一个函数搜索  中的 ,如果目标值存在返回下标,否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/binary-search

2 解法1:左闭右开

使用区间:左闭右开,详细思路见文中代码注释:

class Solution(object):
    def search(self, nums, target):
        # 刚开始的取值区间:[0,len(nums) )
        left,right=0,len(nums)
        while right - left > 1:
            mid = (left+right) //2
            # nums[0]<=...
            if nums[mid] <= target:
                # 遍历后区间改变为:[mid,right)
                left = mid
            else:
                # nums[right-1]...>=nums[mid+1]>=nums[mid] > target
                # 区间调整为:[left,right)
                right = mid
        # 迭代结束后
        # right-left=1,区间为[left, right=left+1)        
        # 即nums[left]就是二分后逼近的点
        # 判断nums[left]是否等于target即可
        return left if nums[left] == target else -1

2 解法2:左闭右闭

class Solution(object):
    def search(self, nums, target):
        # 左右都是闭合区间的写法
        # 刚开始的取值区间:[0,len(nums)-1]
        left,right=0,len(nums)-1
        while left <= right:
            mid = (left+right) //2
            if nums[mid] == target:
                return mid
            # nums[0]<=...
            elif nums[mid] < target:
                # 遍历后区间改变为:[mid+1,right]
                left = mid + 1 
            else:
                # nums[right-1]...>=nums[mid+1]>=nums[mid] > target
                # 区间调整为:[left,mid-1]
                right = mid-1
        # 迭代结束后
        # left - right = 1,此时区间[left,right] = [left,left-1]变为空!
        # 所以返回 -1
        return -1

二分查找,是最基本的分支法的一个应用,面试中被问道的频率很高,同时边界取值特别容易出错,所以单独写为一篇文章,带有详细的注释,希望将来面试能帮助到你一点。觉得还可以,可否点个赞。

算法参考书

如果你的英文还不错,可以学习下面这本算法分析书,几乎从0开始教会你算法的基本概念,数据结构,然后基本的算法设计技术,常用到的比如分治、贪心、动归、回溯、分治定界、随机算法等。

掌握这300页,不管你是软件设计、算法设计、数据分析、爬虫等,基本的算法思想算是掌握了,大家加油!

若有需求,加我微信,备注 算法

浏览 14
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报