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

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

题目描述：

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

进阶：
你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

提示：

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

题目链接

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] result = {-1,-1};
        // 首先二分查找
        if(nums.length == 0) return result; // 没有元素
        int left = 0;
        int right = nums.length - 1;
        if(right == left && target == nums[left]){ // 只有一个元素
            result[0] = 0;
            result[1] = 0;
            return result;
        } 
        while(left<=right){
            int mid = (left+right)/2;
            if(nums[mid] < target){
                left = mid+1;
            }else if(nums[mid] > target){
                right = mid-1;
            }
            // 分别向左和向右扩散
            else{
                int l = mid;
                while(l-1 >= 0 && nums[l - 1] == target){
                    l -- ;
                }
                int r = mid;
                while(r + 1 < nums.length && nums[r + 1] == target){
                    r ++ ;
                } 
                result[0] = l;
                result[1] = r;
                break;
            }
        }
        return result;
    }
}

首先二分查找，如果可以找到target等于数组中的某一个下标为mid的值情况下，继续对其分别向左和向右遍历，找出一个相等区间，然后最终返回即可。