LEETCODE | PYTHON | 34 | 在排序数组中查找元素的第一个和最后一个位置

LEETCODE | PYTHON | 34 | 在排序数组中查找元素的第一个和最后一个位置

1. 题目

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

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

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

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

2. 代码

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        left = 0 
        right = len(nums)-1

        if len(nums) == 0:
            return [-1,-1]

        if target<nums[0]:
            return [-1,-1]
        
        if target>nums[-1]:
            return [-1,-1]

        while left<=right:
            middle = left+(right-left)//2
            if nums[middle]<target:
                left = middle+1
            elif nums[middle]>target:
                right = middle-1
            else:
                break
        
        if nums[middle]!=target:
            return [-1,-1]
        
        index1 = middle
        index2 = middle

        while index1>=1 and nums[index1-1]==target:
            index1 = index1 - 1
        
        while index2<len(nums)-1 and nums[index2+1]==target:
            index2 = index2 + 1

        return [index1,index2]

3. 相关知识

(1) 采用二分查找
(2) 剔除不满足要求的情况，输出[-1,-1]
(3) 先用二分查找找到target的位置，随后利用滑动确定起终点位置