【LeetCode】返回排序数组中值等于某个元素的下标范围:Search for a Range

Given a sorted array of integers, find the starting and ending position of a given target value. Your algorithm’s runtime complexity must be in the order of O(log n).
If the target is not found in the array, return .
For example, Given and target value 8, return .

代码如下:

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        // lower_bound函数找到第一个大于或者等于target的位置
        int left = distance(nums.begin(),lower_bound(nums.begin(),nums.end(),target));
        // upper_bound函数找到第一个大于target的位置
        int right = distance(nums.begin(),prev(upper_bound(nums.begin(),nums.end(),target)));
        // 上述函数可能返回last
        if(nums[left] != target)
            return vector<int> {-1,-1};
        else    
            return vector<int> {left,right};
    }

    template<typename ForwardIterator, typename T>
    ForwardIterator lower_bound(ForwardIterator start, ForwardIterator end, T target) {
        while(start != end) {
            auto mid = next(start, distance(start, end)/2);
            if(*mid < target) start = mid + 1;
            else end = mid;
        }
        
        return start;
    }

    template<typename ForwardIterator, typename T>
    ForwardIterator upper_bound(ForwardIterator start, ForwardIterator end, T target) {
        while(start != end) {
            auto mid = next(start, distance(start, end)/2);
            if(*mid <= target) start = mid + 1; // upper_bound函数仅改行与lower_bound不一样
            else end = mid;
        }
        
        return start;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值