代码题—lower_bound和upper_bound算法

Posted 2020-11-18 eilearn

 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;
}