Leetcode 34 查找元素第一个和最后一个位置

Leetcode 34

力扣

题目：

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

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

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

解题：

二分查找

分析：

该题目可以分解为，寻找第一个等于target的位置，和寻找第一个大于target的位置。简单的二分查找，当mid值等于target时，直接returnmid，而查找第一个位置时，当相等时，需要继续查找左区间，来检查是否还存在和target相等的值。

    def searchRange(self, nums: List[int], target: int) -> List[int]:
        left,right=0,len(nums)-1
        # 寻找到第一个值为target的下标；
        # 当mid的值等于target时，查看mid左区间，继续查找是否存在值为target的num
        # 最终下标为left的num即可能为第一个位置
        def findlower(left:int,right:int,nums:List[int], target: int)->int:
            while left<=right:
                mid = (right-left)//2 + left
                num = nums[mid]
                if num >=target:
                    right = mid-1
                elif num < target:
                    left = mid+1
            return left
        # 寻找到第一个大于target的值
        # 返回right，此时的right则为值等于target的最大位置。
        def findhigh(left:int,right:int,nums:List[int], target: int)->int:
            while left<=right:
                mid = (right-left)//2 + left
                num = nums[mid]
                if num >target:
                    right = mid-1
                elif num <= target:
                    left = mid+1
            return right
        low = findlower(left,right,nums,target)
        high = findhigh(left,right,nums,target)
        if low > high:
            return [-1,-1]
        else:
            return [low,high]