搜索算法-二分查找-34.在排序数组中查找元素的第一个位置和最后一个位置

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]

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

二分查找到一个值（可能左面或右面还有该值），如果左面还有值则将左边界换成左面的那个下标，如果右面还有值则将右边界换成右面的那个下标。

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        return self.binary_search(nums, target, 0, len(nums)-1)

    def binary_search(self, nums, target, left, right):
        if (left > right):
            return [-1, -1]
        mid = (left + right) // 2
        if nums[mid] == target:
            array = [mid, mid]
            left = self.binary_search(nums, target, left, mid - 1)
            right = self.binary_search(nums, target, mid + 1, right)
            # 如果左边不是-1的话，说明左边还有target值，那么就将下界缩小到返回的左值
            if not (left[0] == -1 and left[1] == -1):
                array[0] = left[0]
            if not (right[0] == -1 and right[1] == -1):
                array[1] = right[1]
            return array
        elif target > nums[mid]:
            return self.binary_search(nums, target, mid + 1, right)
        else:
            return self.binary_search(nums, target, left, mid - 1)


# 直接API侠
class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        try:
            left = nums.index(target)
            count = nums.count(target)
            right = left + count - 1
            return [left, right]
        except(ValueError):
            return [-1, -1]