[Leetcode]在排序数组中查找元素的第一个和最后一个位置-二分查找

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

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

题目描述

给定一个按照升序排列的整数数组 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]

解题思路

• 基本二分查找思路：nums[mid]小于target，left++；nums[mid]大于target，right–；直到nums[mid]等于target停止。
• 在基本二分查找基础上，以[left，right)左闭右开为例
  • 如果找左边界，nums[mid]等于target时不马上停止，而是让right等于mid，这样就能找到左边界
  • 如果找右边界，nums[mid]等于target时不马上停止，而是让left等于mid+1，这样就能找到右边界

实现代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int len=nums.size();
        if(len==0||(len==1&&nums[0]!=target))
            return {-1,-1};
        vector<int>res;
        int left=0,right=len; //左闭右开
        while(left<right){ //找左边界
            int mid=(left+right)/2;
            if(nums[mid]>=target) 
                right=mid;
            else if(nums[mid]<target)
                left=mid+1;
        }
        if(left==-1||right==len||nums[right]!=target) //不存在target
            return {-1,-1};
        res.push_back(left);
        right=len;
        while(left<right){ //找右边界
            int mid=(left+right)/2;
            if(nums[mid]>target)
                right=mid;
            else if(nums[mid]<=target)
                left=mid+1;
        }
        res.push_back(right-1);
        return res;
    }
};