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]

提示：

• 0 <= nums.length <= 105
• -109 <= nums[i] <= 109
• nums 是一个非递减数组
• -109 <= target <= 109

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

官方题解
便于理解

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int low = binarySearch(nums,target,true);
        int high = binarySearch(nums,target,false)-1;
        if(low <= high && nums[low] == target && nums[high] == target){
            return new int[]{low,high};
        }
        return new int[]{-1,-1};
    }

    public int binarySearch(int[] nums, int target, boolean low){
        int left = 0,right = nums.length-1,res = nums.length;
        while(left <= right){
            int mid = left + (right - left)/2;
            if(nums[mid] > target || (low && nums[mid] == target)){
                right = mid-1;
                res = mid;
            }else{
                left = mid + 1;
            }
        }
        return res;
    }
}