二分查找【python总结】

将原本是线性时间提升到了对数时间log(N)范围,大大缩短了搜索时间

前提,必须在有序数据中进行查找。

1. 最基本的二分查找

leetcode参考[35]:Search Insert Position

剑指offer:数字在排序数组中出现的次数

def binarySearch(A, target):
    low,high = 0,len(A)-1
    while low <= high:
        mid = low + (high - low) // 2
        if A[mid] == target:
            return mid
        elif A[mid] > target:
            high = mid - 1
        else:
            low = mid + 1
    return -1

其中,有几个要注意的点:

  1. 循环的判定条件是:low <= high
  2. 为了防止数值溢出,mid = low + (high - low)/2
  3. A[mid]不等于target时,high = mid - 1low = mid + 1

2. 查找目标值区域的边界

Leetcode参考[34]: Find First and Last Position of Element in Sorted Array

剑指offer:数字在排序数组中出现的次数

2.1 查找目标值区域的左边界/查找与目标值相等的第一个位置/查找第一个不小于目标值数的位置

A = [1,3,3,5, 7 ,7,7,7,8,14,14]
target = 7
return 4

def binarySearchLowerBound(A,  target):
    low,high = 0,len(A)-1
    while low <= high:
        mid = low + (high - low) // 2
        if target <= A[mid]:
            high = mid - 1
        else:
            low = mid + 1
    if low < A.length and A[low] == target:
        return low
    else
        return -1

2.2 查找目标值区域的右边界/查找与目标值相等的最后一个位置/查找最后一个不大于目标值数的位置

A = [1,3,3,5,7,7,7, 7 ,8,14,14]
target = 7
return 7

def binarySearchUpperBound(A,  target):
    low,high = 0,len(A)-1
    while low <= high:
        mid = low + (high - low) // 2
        if target >= A[mid]:
            low = mid + 1
        else:
            high = mid - 1
    if high >-1 and A[low] == target:
        return high
    else:
        return -1

此题以可变形为查找第一个大于目标值的数/查找比目标值大但是最接近目标值的数,我们已经找到了最后一个不大于目标值的数,那么再往后进一位,返回high + 1,就是第一个大于目标值的数。

2.3 查找最后一个小于目标值的数/查找比目标值小但是最接近目标值的数

此题以可由第 2 题变形而来,我们已经找到了目标值区域的下(左)边界,那么再往左退一位,即low - 1,就是最后一个小于目标值的数。其实low - 1也是退出循环后high的值,因为此时 high刚好等于low - 1,它小于low,所以 while 循环结束。我们只要判断high是否超出边界即可。

A = [1,3,3, 5 ,7,7,7,7,8,14,14]
target = 7
return 3

def binarySearchLowerBound2(A,  target):
    low,high = 0,len(A)-1
    while low <= high:
        mid = low + (high - low) // 2
        if target > A[mid]:
            low = mid + 1
        else:
            high = mid - 1
    if high >-1:
        return high
    else:
        return -1

2.4 查找第一个大于目标值的数/查找比目标值大但是最接近目标值的数

此题以可由第 3 题变形而来,我们已经找到了目标值区域的上(右)边界,那么再往右进一位,即high + 1,就是第一个大于目标值的数。其实high + 1也是退出循环后low的值,因为此时 low刚好等于high + 1,它大于high,所以 while 循环结束。我们只要判断low是否超出边界即可。

A = [1,3,3,5,7,7,7,7, 8 ,14,14]
target = 7
return 8

def binarySearchUpperBound2(A,  target):
    low,high = 0,len(A)-1
    while low <= high:
        mid = low + (high - low) // 2
        if target >= A[mid]:
            low = mid + 1
        else:
            high = mid - 1
    if high < len(A):
        return high
    else:
        return -1

3. 在旋转数组中查找最小元素

3.1 查找旋转数组的最小元素(假设不存在重复数字)

LeetCode[153]: Find Minimum in Rotated Sorted Array
Input: [3,4,5,1,2]
Output: 1

def findMin(nums):
    if len(nums) == 0
        return -1
    left,right = 0,len(nums) - 1
    while left < right:
        mid = left + (right - left) // 2
        if nums[mid] > nums[right]:
            left = mid + 1
        else:
            right = mid
    return nums[left]

注意这里和之前的二分查找的几点区别:

  1. 循环判定条件为left < right,没有等于号
  2. 循环中,通过比较nums[left]与num[mid]的值来判断mid所在的位置:
  • 如果nums[mid] > nums[right],说明前半部分是有序的,最小值在后半部分,令left = mid + 1
  • 如果nums[mid] <= num[right],说明最小值在前半部分,令right = mid

最后,left会指向最小值元素所在的位置。

3.2 查找旋转数组的最小元素(存在重复项)

LeetCode[154]: Find Minimum in Rotated Sorted Array II
剑指offer:旋转数组的最小数字
Input: [2,2,2,0,1]
Output: 0

def findMin(nums):
    if len(nums) == 0
        return -1
    left,right = 0,len(nums) - 1
    while left < right:
        mid = left + (right - left) // 2
        if nums[mid] > nums[right]:
            left = mid + 1
        elif nums[mid] < nums[right]:
            right = mid
        else:
            right-=1
    return nums[left]

和之前不存在重复项的差别是:当nums[mid] == nums[right]时,我们不能确定最小值在 mid的左边还是右边,所以我们就让右边界减一。

4. 在旋转排序数组中搜索

4.1 在旋转排序数组中搜索并返回目标元素的下标(不考虑重复项)

LeetCode[33]: Search in Rotated Sorted Array

法一:

  • 先利用方法 3.1 查找数组中的最小元素,即确定分界点的位置
  • 把旋转的数组当成偏移,用(offset + mid) % len来求真实的 mid 的位置。
  • 然后用二分查找来定位目标值
def search(self, nums: List[int], target: int) -> int:
    lo,hi=0,len(nums)-1
    while lo<hi:
        mid=(lo+hi)//2
        if nums[mid]>nums[hi]:
            lo=mid+1
        else:
            hi=mid 
    offset =lo #lo==hi is the index of the smallest value
    lo,hi=0,len(nums)-1
    while lo<=hi:
        mid=lo+(high-lo)//2
        realmid=(mid+offset)%len(nums)
        if nums[realmid]==target:
            return realmid
        if nums[realmid]<target: 
            lo=mid+1
        else:
            hi=mid-1
    return -1

法二:其实没有必要找到旋转数组的分界点,对于搜索左侧还是右侧我们是可以根据mid跟high的元素大小来判定出来的,直接根据target的值做二分搜索就可以了。

def search(self, nums: List[int], target: int) -> int:
    if len(nums) == 0:
        return -1
    left,right=0,len(nums)-1
    while left<=right:
        mid=left+(right-left)//2
        if nums[mid] == target:
            return mid
        elif nums[left] <= nums[mid]:
            if target < nums[mid] and target >= nums[left]:
                right = mid - 1
            else:
                left = mid + 1
        elif nums[mid] <= nums[right]:
            if target > nums[mid] and target <= nums[right]:
                left = mid + 1
            else:
                right = mid - 1
    return -1

4.2 在旋转排序数组中搜索并返回目标元素的下标(考虑重复项)

LeetCode: Search in Rotated Sorted Array II

def search(self, nums: List[int], target: int) -> int:
    if len(nums) == 0:
        return -1
    left,right=0,len(nums)-1
    while left<=right:
        mid=left+(right-left)//2
        if nums[mid] == target:
            return mid
        elif nums[mid] > nums[right]:
            if target < nums[mid] and target >= nums[left]:
                right = mid 
            else:
                left = mid + 1
        elif nums[mid] < nums[right]:
            if target > nums[mid] and target <= nums[right]:
                left = mid + 1
            else:
                right = mid
        else:
            right-=1
    return -1

5. 二维数组中的查找

剑指offer:二维数组中的查找

二维数组是有序的,从右上角来看,向左数字递减,向下数字递增。因此可以利用二分查找的思想,从右上角出发:

  • 当要查找数字比右上角数字大时,下移;
  • 当要查找数字比右上角数字小时,左移;

【参考】
作者:繁著
链接:https://www.jianshu.com/p/0f823fbd4d20

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,530评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 86,403评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,120评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,770评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,758评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,649评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,021评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,675评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,931评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,659评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,751评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,410评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,004评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,969评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,042评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,493评论 2 343