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

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

难度中等
给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target，返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1：
输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
示例 2：
输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
示例 3：
输入：nums = [], target = 0
输出：[-1,-1]
1
2
3
4
5
6
7
8
9

提示：

• 0 <= nums.length <= 10(5)
• -10(9) <= nums[i] <= 10(9)
• nums 是一个非递减数组
• -10(9) <= target <= 10(9)

思路

• 关键步骤，与二分查找不同
  • if(target**<=**nums[mid]&&mid-1>=0) binary_search(nums,target,l,mid-1);
    • target==nums[mid]时，不结束循环，继续进入
    • 两个if可以同时进入，寻找界限
    • 设置限制，防止mid-1,mid+1越界

代码

class Solution {
public:
    int left=-1,right=-1;
    vector<int> searchRange(vector<int>& nums, int target) {
        if(nums.size()==0)  return vector<int>{-1,-1};
        binary_search(nums,target,0,nums.size());
        return vector<int>{left,right};
    }
    void binary_search(vector<int>& nums,int target,int l ,int r){
        int mid=(r-l)/2+l;
        //printf("%d-%d\n",l,r);
        if(target==nums[mid]){
            if(left>mid||left<0)  left=mid;
            if(right<mid||right<0)   right=mid;
        }
        if(l>=r) return;
        if(target<=nums[mid]&&mid-1>=0)  binary_search(nums,target,l,mid-1);
        if(target>=nums[mid]&&mid+1<nums.size())   binary_search(nums,target,mid+1,r);
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20