BinarySearch 经典二分查找算法

前言

二分查找算是最经典也最入门的算法了,大一新生刚学C语言就开始写,但是看似简单的二分算法,想要考虑周全写得完美也是要费点功夫。
比如java库里的二分查找,一个溢出的bug还存在了很久,更有号称90%程序员写不出无BUG的二分查找程序,夸张了。。。

自我分析

先来看看我平时最喜欢用的二分查找代码

int binarysearch(int nums[], int low, int high, int key){
    while(low<=high){
        int mid = low + (high - low) / 2;
        if(nums[mid] == key)
            return mid;
        else if(nums[mid]>key)
            high = mid - 1;
        else
            low = mid + 1;
    }
    return low;
}

这里面几个注意的点:

  1. 我老老实实取的两头闭区间,求中位数也防止了溢出
  2. 终结条件为搜索空间为空,也就是low>high
  3. 返回值完全不需要纠结,直接返回低位 low,当然除非要求找不到返回-1

慢放整个搜索过程,实际上是一个维护low的过程:

low为0开始,只在中位数遇到确定小于目标时才前进,并且永不后退。low一直朝着第一个目标的位置接近,直到到达。

这代码一直用的挺好的,但其实有一个前提条件:没有重复值
如果有重复值的话,那么返回的位置就不一定是哪个了,虽然也是目标的位置,但是这个时候我们往往会要求返回第一个大于等于目标值的位置。
那么具体该怎么办呢?其实很简单,当取到等于目标值的时候,之前是直接return,现在我们改为继续减小high来缩小搜索空间。代码如下:

int binarysearch_lower_bound(int nums[], int low, int high, int key){
    while(low<=high){
        int mid = low + (high - low) / 2;
        if(nums[mid]>=key)
            high = mid - 1;
        else
            low = mid + 1;
    }
    return low;
}

其实这就对应C++里的lower_bound()函数了

Classical Code

参考 ClassicalCode/BinarySearch.cpp
上面的版本区间为左闭右开,且边界条件改为while(lo < hi),对应的hi = mid ;我还不怎么适应,下面给出我的版本(左右闭区间)以及递归实现

//二分查找之找到nums[lo, hi]大于等于key值的第一个元素位置
int binarysearch_lower_bound(int nums[], int low, int high, int key){
    while(low<=high){
        int mid = low + (high - low) / 2;
        if(nums[mid]>=key)
            high = mid - 1;
        else
            low = mid + 1;
    }
    return low;
}
//递归版本
int binarysearch_firstoccur(int nums[], int low,int high, int key){
    if (low > high)
        return low;
    int mid = low + (high - low) / 2;
    if(nums[mid]>=key)
        return binarysearch_firstoccur(nums, low, mid-1, key);
    if(nums[mid] < key)
        return binarysearch_firstoccur(nums, mid + 1, high, key);
}

//二分查找之找到严格大于key值的第一个元素
int binarysearch_upper_bound(int nums[], int low, int high, int key){
    while(low<=high){
        int mid = low + (high - low) / 2;
        if(nums[mid]>key)
            high = mid - 1;
        else
            low = mid + 1;
    }
    return low;
}

你可能感兴趣的:(BinarySearch 经典二分查找算法)