打开APP
userphoto
未登录

开通VIP,畅享免费电子书等14项超值服

开通VIP
面试题_二分查找及其变形 - 糖拌咸鱼 - 博客园

   二分查找常用来查找指定有序集合中元素的位置,思路和代码都比较简单,所以大家都很熟练。二分查找貌似很多公司在面试或笔试的时候都会多少涉及到,经常会让你在纸上直接写代码,所以平常只知道原理而从来不自己写的人,可能不会太快写出来,或是代码有点小漏洞,所以经常敲敲常见的数据结构和代码还是很必要的。

很多问题,只要涉及到有序的数据结构,就可以往二分查找的思路上靠。 下面的代码,可能不是很完善,仅供参考。

经典二分查找的代码,在有序数组里面查找是否存在某个元素,如果存在,则返回相应元素所在索引号:

int Find(int arr[], int key,int length)
{
assert(arr
!=NULL&&length>0);
int low=0,high=length,mid;
while(low<=high)
{
mid
=(low+high)/2;
if(arr[mid]==key) return mid;
else
{
if(arr[mid]>key) high=mid-1;
else low=mid+1;
}
}
return-1;
}

变形1:

问题来了:在循环有序数组中查找指定元素,也就是说在类似这样的{12,16,18,20,41,100,1,4,6,9}数组中查找指定的元素。

  分析一下,这里所说的循环有序数组,就是把一个有序数组从某个(未知)位置处截为两段,把前一段放到后一段的后面(数组里的元素还是有序的,只不过最小值不一定是数组的第一个元素,而可能是其中的任何一项,从它开始逐项递增,到数组的最后一个元素时再回到第一个元素)。
显然传统的二分法已经无法直接使用了,但考虑一下,如果已经知道分界点位置,那问题就简单多了,只要先判断一下待查元素是在分界点的左侧还是右侧,然后直接对那一侧的半个数组使用二分查找。

  那么重点就是判定待测元素在分界点的左侧还是右侧的问题了,可以发现每次取mid后,就会形成两种情况的子序列。一种情况是类似{4,6,9},他是一个正常有序的子集合,另一种情况是类似{12,16,18,20,41,100,1}的与源问题类似结构的相对复杂的子集合。显然第一种情况是简单的,那么判定待测元素在分界点的简单一侧会比较容易。

  第一种情况(arr[mid]>=arr[low]):当key<=arr[mid]&&key>=arr[low]时,待测元素肯定会在mid的左侧;其他情形则会在mid的右侧。

  第二种情况(arr[mid]<arr[low]):当key<=arr[low]&&key>=arr[mid]时,待测元素肯定会在mid的右侧,其他情形则会在mid的左侧。

  上面两个子条件的选择比较重要。

 最后给出代码:

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

 

变形2:假如集合中的元素有重复,要找到key首次出现的位置。

int find(int * arr , int length, int key){    int low = 0,high = length-1,mid;    while(low <= high){        mid = (low + high)/2;        if (arr[mid] < key){            low = mid + 1;        }else if(arr[mid] > key){            high = mid - 1;        }else{            if (mid == 0) return mid;            if (mid >= 1 && arr[mid-1] != key) return mid;            else high = mid - 1;        }    }    return -1;}

变形3:在一个有序的数组里,查找不小于key的最小的数的索引,如果key值存在,则返回key第一次出现的位置,如果不存在则返回-1。(类似的查找比key小的最大的数) 

#include<iostream>using namespace std;int find(int * arr, int low, int high, int key){    if(arr[low] >= key) return low;    if (low > high) return -1;    int mid = (low + high) / 2;    if (arr[mid] < key) return find(arr,mid+1,high,key);    else if(arr[mid] >= key){        if(mid >= low && arr[mid-1]>=key)            return find(arr,low, mid-1, key);        return mid;    }}int main(){    int arr[] = {1,3,3,3,3,3,6,6,7,9,11,12};    int length = sizeof(arr) / sizeof(int);    cout<<'KEY'<<'\t'<<'索引'<<'\t'<<'对应值'<<endl;    for(int i=0;i<15;i++){        int idx = find(arr,0,length-1,i);        if (idx >= 0)            cout<<i<<'\t'<<idx<<'\t'<<arr[idx]<<endl;        else cout<<i<<'\t'<<idx<<'\t'<<'None'<<endl;    }    return 0;}

变形4:在一个有序的数组里,数据里面元素可能有重复的,查找指定key所在的索引范围。

例如:int arr[] = {1,2,2,2,2,3,3,3,3,3,6,6,7,9,11,12}; 查找3的话,应该返回(5,9)

分析一下: 一个思路是,先随意找到一个key所在的索引i,然后从i的左右两侧查找,直到左右两侧遇到不等于key的值为止,然后返回两边的索引号。

但是,这样存在一个问题,考虑一下如果一个数组是{1,2,3,3,3,3,3,3,3,4},这样按照上面的想法就很扯淡了,因为与key相等的值太多了,比较代价太大。

于是可以考虑,用两次二分查找,找两个边界,这里我们可以使用变形3中所写好的find函数,我们去找比key+1大的最小的索引right,以及比key大的最小值得索引left。

返回结果,即为<left,right-1>

这里有个问题必须指出,在变形3我们写的函数中,如果key存在则返回key第一次出现的位置。

 

int find(int * arr, int low, int high, int key){    if(arr[low] >= key) return low;    if (low > high) return -1;    int mid = (low + high) / 2;    if (arr[mid] < key) return find(arr,mid+1,high,key);    else if(arr[mid] >= key){        if(mid >= low && arr[mid-1]>=key)            return find(arr,low, mid-1, key);        return mid;    }}pair<int,int> findRange(int * arr, int length, int key){    pair<int,int> res(-1,-1);    int l = find(arr, 0,length-1, key);    if(l<0 || arr[l] != key) return res;    res.first = l;    int r = find(arr, 0, length-1, key+1);    if(r-1<0 || arr[r-1] != key) return res;    res.second = r-1;    return res;}int main(){    int arr[] = {1,2,2,2,2,3,3,3,3,3,6,6,7,9,11,12};    int length = sizeof(arr) / sizeof(int);    for (int i=0;i<15;i++){        pair<int,int> res = findRange(arr, length,i);        cout<<i<<'\t'<<res.first<<'\t'<<res.second<<endl;    }    return 0;}

 

 

 

本站仅提供存储服务,所有内容均由用户发布,如发现有害或侵权内容,请点击举报
打开APP,阅读全文并永久保存 查看更多类似文章
猜你喜欢
类似文章
【热】打开小程序,算一算2024你的财运
Java实现简单的二分法查找int数组中的某值
各种常见排序算法大汇总
必学十大经典排序算法,看这篇就够了
七大排序算法(冒泡,选择,插入,二分法排序,希尔,快速,合并,堆排序)的java实现(14/8/3更新加入二分排序)
第十三天-总结
找到 K 个最接近的元素
更多类似文章 >>
生活服务
热点新闻
分享 收藏 导长图 关注 下载文章
绑定账号成功
后续可登录账号畅享VIP特权!
如果VIP功能使用有故障,
可点击这里联系客服!

联系客服