0
点赞
收藏
分享

微信扫一扫

二分查找——妙用STL(lower_bound和upper_bound)

小布_cvg 2022-01-06 阅读 41

二分查找(在这就不过多赘叙,大伙可以自行了解)

       本文主要讲lowe_bound和upper_bound这两个底层皆基于二分查找,当然STL标准库中还有 equal_range() 和 binary_search() 也是基于二分查找实现的。读者可自行了解!

C++ lower_bound()函数

       lower_bound() 函数用于在指定区域内查找不小于目标值的第一个元素。也就是说,使用该函数在指定范围内查找某个目标值时,最终查找到的不一定是和目标值相等的元素,还可能是比目标值大的元素。
       以下为其常用的两种用法:
       //在 [first, last) 区域内查找不小于 val 的元素
ForwardIterator lower_bound (ForwardIterator first, ForwardIterator last,const T& val);

       //在 [first, last) 区域内查找第一个不符合 comp 规则的元素
ForwardIterator lower_bound (ForwardIterator first, ForwardIterator last, const T& val, Compare comp);

C++ upper_bound()函数

       lower_bound()函数 用于在指定范围内查找大于目标值的第一个元素。
       以下为其常用的两种用法(与lower_bound类似):
       //查找[first, last)区域中第一个大于 val 的元素。
ForwardIterator upper_bound (ForwardIterator first, ForwardIterator last,const T& val);

       //查找[first, last)区域中第一个不符合 comp 规则的元素
ForwardIterator upper_bound (ForwardIterator first, ForwardIterator last, const T& val, Compare comp);

下面以LeetCode简单算法题为例:

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。
如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

       常规解法:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int length = nums.size();
        int left=0,right=length 1;
        while(left<=right){
            int mid=left+(right-left)/2; // 这里一定要注意不仅仅是尾部减去首部除以二 还需加上尾部下标
            if(nums[mid]<target)
                left=mid+1;
            else right=mid-1;
        }
        return left;
    }
};

在这里插入图片描述

       STL解法:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        return lower_bound(nums.begin(),nums.end(),target) - nums.begin();
    }
};

在这里插入图片描述

举报

相关推荐

0 条评论