hot100-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]

思路

利用二分思想先找其左边界，再找其右边界即可，注意找左边界的时候，由右侧逼近；找右边界的时候，由左侧逼近，即可。

代码

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] res = new int[]{-1,-1};
        if(nums==null || nums.length==0)return res;
        res[0] = findEdge(nums,target,true);
        res[1] = findEdge(nums,target,false);
        return res;
    }

    private int findEdge(int[] nums,int target,boolean findLeftEdge){
        int res = -1;
        int low = 0;
        int high = nums.length-1;
        int mid;
        while(low<=high){
            mid = low + (high-low)/2;
            if(nums[mid]>target){
                high = mid-1;
            }else if(nums[mid]<target){
                low = mid+1;
            }else{
                res = mid;
                if(findLeftEdge){
                    high = mid-1;
                }else{
                    low = mid+1;
                }
            }
        }
        return res;
    }
}

//利用二分思想先找其左边界，再找其右边界即可，注意找左边界的时候，由右侧逼近；找右边界的时候，由左侧逼近，即可。