Leetcode 34.在排序数组中查找元素的第一个和最后一个位置

在排序数组中查找元素的第一个和最后一个位置

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。

如果数组中不存在目标值，返回 [-1, -1]。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8

输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6

输出: [-1,-1]

 

class Solution {
public:
    vector<int>searchRange(vector<int>& nums,int target){
        vector<int> r(2,-1);
        if(!nums.size()) return r;
        int l=0,h=nums.size()-1,m=0;
        if(target<nums[l] || nums[h]<target) return r;
        while(l<=h){
            m=(l+h)/2;
            if(nums[m]==target){
                int i=m;
                while(nums[i]==target&&i<nums.size()) ++i;
                r[1]=i-1;
                while(nums[m]==target&&m>=0) --m;
                r[0]=m+1;
                return r;
            }else if(nums[m]<target) l=m+1;
            else h=m-1;
        }
        return r;
    }
};