如何找到二分查找中目标元素第一次出现和最后一次出现的位置

http://blog.chinaunix.net/uid-27103408-id-3761907.html


关于二分查找有些想不清楚的边界条件问题,在这篇博文里讲得很清楚了。如果能深刻理解那么将会十分受用。


大家对于二分查找并不陌生,一般意义上的二分查找,往往返回给我们的是目标元素在排序数组中出现的一个随机的位置,但是在很多时候,我们却是需要目标元素的第一个和最后一个位置,才有意义。举个例子来说,我们要求得目标元素在排序数组中出现的次数,假如利用一般的方法,逐个比较目标元素和数组元素,时间复杂度O(n),不能够令我们满意,既然数组是排序的我们很容易想到二分查找,在这里我们能不能使用二分查找的算法呢,答案是肯定的。只要我们能够利用二分查找找到目标元素出现的第一个和最后一个位置,就能够求得它出现的次数。我们如何来求得目标元素出现的第一个和最后一个位置呢,其实很简单,我们只需要对于二分查找的退出条件,做一个简单的设定就能得到我们理想的结果哦!
下面我们来看一下代码:

int GetFirstK(int *a, int _left, int _right, int dest)
{
    if(_left > _right || a == NULL)
    {
        return 0;
    }
    int temp = 0;
    int left = _left;
    int right = _right;
    while(left < right)
    {
        temp = (left + right) >> 1;
        if(a[temp] < dest)
        {
            left = temp + 1;
        }
        else
        {
            right = temp;
        }
    }
    return left;
}

在这里跟一般的二分查找的代码相比,仅仅是判断语句上做了一点细微的变化,序列是递增排列的,当中间值小于目标元素的时候,目标元素在序列的右边:left = temp + 1;
其余的情况目标值在序列的左边:right = temp;我们要查找的第一个目标元素的位置,一般的情况就是目标元素存在多个情况,由上述的两个判断条件,我们可以知道,如果查找到了目标元素,并且该目标元素不是第一个的时候,此时left<right,判断继续进行。我们这里设置的判断条件保证了,当中间值小于目标元素的时候,left一定会向右移动,同时如果中间值恰好等于目标元素的时候,right能够不动,最终序列结束的时候,left=right=目标元素的第一次出现的位置!整个过程大家可以设定一个例子在纸上模拟一下:

去最后一次出现的位置,道理也是类似的:

int GetUpK(int *a, int _left, int _right, int dest)
{
    if(_left > _right || a == NULL)
    {
        return 0;
    }
    int temp = 0;
    int left = _left;
    int right = _right;
    while(left < right)
    {
        temp = (left + right + 1) >> 1; //保证取到中间靠后的位置
        if(a[temp] > dest)
        {
            right = temp - 1;
        }
        else
        {
            left = temp;
        }
    }
    return right;
}

大家可以看出,跟我们取第一个元素时候的判断条件恰好相反,而两种情况处理的方式我们可以归结为以下两句话:


1、当我们要找到目标元素出现的第一个位置时候:当中间值大于等于目标元素的时候,我们要保留当前中间值的位置,并且在左边继续查找。
这句话用条件语句表述就是:
if (a[mid] >= dest)
    right = mid;


2、当我们要找目标元素出现的最后一个位置的时候:当中间值小于等于目标元素的时候,我们要保留中间值的位置,并且在右边继续查找。
if(a[mid] <= dest)
    left = mid;

好的,下面是二分查找的递归实现代码: ```python def binary_search(arr, target, left, right): if left > right: return -1 mid = (left + right) // 2 if arr[mid] == target: if mid == 0 or arr[mid-1] != target: # 判断是否是第一次出现 return mid else: return binary_search(arr, target, left, mid-1) elif arr[mid] > target: return binary_search(arr, target, left, mid-1) else: return binary_search(arr, target, mid+1, right) ``` 其,arr是已排序的数组,target是要查找的目标元素,left和right分别是当前查找范围的左右边界。 在二分查找,每次取mid,如果mid等于目标元素target,则找到目标元素。但是,我们还要判断当前mid是不是第一次出现位置,如果是,则直接返回mid;否则,继续在左半部分查找。 如果mid大于目标元素target,则说明目标元素在左半部分,我们继续在左半部分进行查找。 如果mid小于目标元素target,则说明目标元素在右半部分,我们继续在右半部分进行查找。 如果left > right,则说明目标元素不在数组,返回-1表示未找到最后,我们可以调用binary_search函数来查找元素第一次出现位置: ```python arr = [1, 2, 3, 3, 3, 4, 5, 6] target = 3 index = binary_search(arr, target, 0, len(arr)-1) print(index) ``` 输出结果为:2 注意,这里的index表示的是目标元素在数组第一次出现位置,如果目标元素不在数组,则输出-1。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值