力扣刷题打卡day8 在排序数组中查找元素的第一个和最后一个位置34

题目描述

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

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

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

思路一：二分查找

• 先定位目标是否存在，再向两端扩展确定返回的左右索引

 class Solution{
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int>res;
        if(nums.empty()){
            res.push_back(-1);
            res.push_back(-1);
            return res;
        }
        int begin=0,end=(int)nums.size();
        int mid=(begin+end)/2;
        while(true){
            if(target==nums[mid]){//向左右扩张
                int left=mid,right=mid;
                while(left-1>=0&&nums[left-1]==target){
                    left--;
                }
                while(right+1<nums.size()&&nums[right+1]==target){
                    right++;
                }
                res.push_back(left);
                res.push_back(right);
                return res;
            }
            if(nums[mid]<target){
                begin=mid;
            }else{
                end=mid;
            }
            if((begin+end)/2==mid){
                res.push_back(-1);
                res.push_back(-1);
                return res;
            }
            mid=(begin+end)/2;
        }
        return res;
    }
};