二分查找求解equal_range区间

总觉得二分查找很简单,可实现起来却发现一个小错误会导致程序结果出错,本题是解决眼高手低的利题

eaual_range返回的一对指针,指向第一个等于key值得元素和指向第一个大于该值的元素。因此,除了常规的非重复元素的二分查找(重复元素可能查找到任意一个值),还可以实现equal_range的方式考察。

先看常规的二分查找

int binsearch(int arr[], int left, int right, int key)
{
    if(arr == nullptr)
        return -1; 
    while(left <= right)
    {
        int mid = left + ((right - left) >> 1);
        if(arr[mid] < key)
            left = mid + 1;
        if(arr[mid] > key)
            right = mid - 1;
        else
            return mid; 
    }   
    return -1;
}

查找元素第一次出现的下标

int binsearch_min(int arr[], int left, int right, int key)
{
    if(!arr)    
        return -1;
    while(left < right)
    {
        int mid = left + ((right-left)>>1);
        if(arr[mid] < key)
        {
            left = mid+1;
        }else
        {
            right = mid;
        }
    }
    if(arr[left] == key)   
        return left;
    return -1;
}

返回刚好大于key的元素下标

int binsearch_justgreat(int arr[], int left, int right, int key)
{
    if(!arr)    
        return -1;
    while(left < right)
    {
        int mid = left + ((right-left)>>1);
        if(arr[mid] <= key)
        {
            left = mid + 1;
        }else
        {
            right = mid;
        }
    }
    if(arr[right] > key) 
        return right;
    return -1;
}

你可能感兴趣的:(算法)