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

1. 题目大意

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

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

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

2. 思路分析

示例 1：
输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

首先题目说是一个顺序排列的数组，让查找一个目标值，且复杂度为O(logn)，熟不熟悉？是不是二分查找的思路。
里面有一点值得注意，数组中的元素是有重复的，在找到目标值后，需要查找其前面和后面的元素，最终返回一个包含所有目标值的下标。

3. 代码示例

Java版本

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int i=0, j=nums.length-1; 
        int l = j;

        while(i<=j){
            int mid = (i+j) / 2;
            if(nums[mid] == target){
                int left = mid, right = mid;
                while(left-1 >= 0 && nums[left - 1] == target)
                    left--;
                while(right+1<=l && nums[right + 1] == target)
                    right++;
                return new int [] {left, right};
            }
            else if(nums[mid] > target)
                j = mid - 1;
            else
                i = mid + 1;
        }
        return new int [] {-1, -1};
    }
}

Python版本

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        i, j= 0, len(nums)-1
        l = j
        res = [-1, -1]
        while i<=j:
            mid = (i+j) // 2
            if nums[mid] == target:
                left = right = mid 
                # 往左查找
                while left-1 >= 0 and nums[left - 1] == target:
                    left -= 1
                # 往右查找
                while right+1<=l and nums[right + 1] == target:
                    right += 1
                # 返回结果
                return [left, right]
            elif nums[mid] > target:
                j = mid - 1
            else:
                i = mid + 1
        return res