美文网首页
BinarySearch 经典二分查找算法

BinarySearch 经典二分查找算法

作者: leon4ever | 来源:发表于2017-12-31 18:53 被阅读98次

    前言

    二分查找算是最经典也最入门的算法了,大一新生刚学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 经典二分查找算法

          本文链接:https://www.haomeiwen.com/subject/pysagxtx.html