二分元素第一次出现的位置和最后一次出现的位置

最新推荐文章于 2024-12-05 19:27:25 发布

原创最新推荐文章于 2024-12-05 19:27:25 发布 · 1.9k 阅读

2 ·

CC 4.0 BY-SA版权

Acm 专栏收录该内容

36 篇文章

订阅专栏

本文介绍了一种使用二分查找法实现的上界和下界搜索算法，旨在为STL中的lower_bound()和upper_bound()提供替代方案。通过自定义函数binsearch_low()和binsearch_up()，可以在有序数组中找到指定元素的第一个或最后一个出现位置。

stl里面封装了挺不错的lower_bound()和upper_bound()。但是有时候条件的判断又得自己写，所以摸索着写了两个函数，还是比较习惯 (low<=high) 的写法

//上界：
int binsearch_low(int lef, int rig,int key,int a[])
{
    int ans = -1;
    int low = lef,high = rig;
    while(low <= high){
        int mid = (low + high)/2;
        if(key == a[mid]){
            ans = mid;
            high = mid - 1;
        }
        else if(key < a[mid])
            high = mid - 1;
        else
            low = mid + 1;
    }
    return ans;
}



//下界：
int binsearch_up(int lef, int rig,int key,int a[])
{
    int ans = -1;
    int low = lef,high = rig;
    while(low <= high){
        int mid = (low + high)/2;
        if(key == a[mid]){
            ans = mid;
            low = mid + 1;
        }
        else if(key < a[mid]){
            high = mid - 1;
        }
        else{
            low = mid + 1;
        }
    }
    return ans;
}