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

题目描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。你的算法时间复杂度必须是 O(log n) 级别。如果数组中不存在目标值，返回 [-1, -1]。

示例一

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

示例二

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

思路

用两次二分即可解决。
第一次二分查找第一个小于等于target的位置，若该位置不等于target，则返回[-1, -1]。
第二次二分查找第一个大于target的位置。注意需要特判边界。

代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.empty()) return vector<int>({-1,-1}) ;
        vector<int>ans;
        int l=0,r=nums.size()-1;
        while(l<r)
        {
            int mid=l+r>>1;
            if(nums[mid]>=target)
                r=mid;
            else
                l=mid+1;
        }
        if(nums[r]!=target) return vector<int>({-1,-1}) ;
            ans.push_back(r);
        l=0,r=nums.size()-1;
        while(l<r)
        {
            int mid=l+r+1>>1;
            if(nums[mid]<=target)
                l=mid;
            else
                r=mid-1;
        }
        ans.push_back(r);
        return ans;       
    }
};