代码题—lower_bound和upper_bound算法

时间:2021-02-08 20:13:47

 1、lower_bound:查找序列中的第一个出现的值大于等于val的位置

  这个序列中可能会有很多重复的元素,也可能所有的元素都相同,为了充分考虑这种边界条件,STL中的lower_bound算法总体上是才用了二分查找的方法,但是由于是查找序列中的第一个出现的值大于等于val的位置,所以算法要在二分查找的基础上做一些细微的改动。

     首先是我修改数据结构课本上的二分查找实现的lower_bound算法:

int lower_bound(vector<int> array, int size, int key)
{
    int low = 0, high = size - 1;
    int mid, pos = 0;  //需要pos记录位置
    while (low < high)
    {
        mid = (low + high) / 2;
        if (array[mid] < key)  //若中位数的值小于key的值,我们要在右边子序列中查找,这时候pos可能是右边子序列的第一个
        {
            low = mid + 1;
            pos = low;
        }
        else
        {
            high = mid;  //若中位数的值大于等于key,我们要在左边子序列查找,但有可能middle处就是最终位置,所以我们不移动last
            pos = high;
        }
    }
    return pos;
}

 

2、upper_bound:返回的是最后一个大于等于val的位置,也是有一个新元素val进来时的插入位置。

int upper_bound(vector<int> array, int size, int key)
{
    int low = 0, high = size - 1;
    int mid, pos = 0;
    while (low < high)
    {
        mid = (low + high) / 2;
        if (array[mid] <= key)
        {
            low = mid + 1;
            pos = low;
        }
        else
        {
            high = mid;
            pos = high;
        }
    }
    return pos;
}